Skip to content

@pytorch-tpu pytorch-tpu

Change the repository type filter

All

    Repositories list

    12 repositories

    • diffusers

      Public
      Modification to run inference of Stable Diffusion models from HF. Will be included in the tutorial.
      Python
      •
      Apache License 2.0
      •2•2•0•1•Updated Nov 19, 2024Nov 19, 2024

    • run_torchbench

      Public
      This repo contains a list of stand alone scripts for running torchbench models one by one.
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •2•0•0•1•Updated Nov 19, 2024Nov 19, 2024

    • transformers

      Public
      🤗 Transformers: State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch.
      Python
      •
      Apache License 2.0
      •27k•13•2•5•Updated Nov 10, 2024Nov 10, 2024

    • accelerate

      Public
      🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support
      Python
      •
      Apache License 2.0
      •971•0•0•0•Updated Oct 17, 2024Oct 17, 2024

    • llama

      Public
      Inference code for LLaMA models
      Python
      •
      Other
      •9.6k•20•4•8•Updated May 30, 2024May 30, 2024

    • llm-foundry

      Public
      LLM training code for Databricks foundation models
      Python
      •
      Apache License 2.0
      •530•0•0•0•Updated May 30, 2024May 30, 2024

    • composer

      Public
      Supercharge Your Model Training
      Python
      •
      Apache License 2.0
      •420•0•0•0•Updated May 29, 2024May 29, 2024

    • ray-llama

      Public
      Example for running Llama 2 on Ray with TPUs
      Python
      •
      Apache License 2.0
      •3•10•1•0•Updated Nov 28, 2023Nov 28, 2023

    • stable-diffusion

      Public
      A latent text-to-image diffusion model
      Jupyter Notebook
      •
      Other
      •10k•7•0•0•Updated Sep 11, 2023Sep 11, 2023

    • training

      Public
      Reference implementations of MLPerf™ training benchmarks
      Python
      •
      Other
      •561•3•0•0•Updated Sep 8, 2023Sep 8, 2023

    • examples

      Public
      This repository contains example code to build models on TPUs
      Shell
      •
      Apache License 2.0
      •11•30•0•0•Updated Feb 17, 2023Feb 17, 2023

    • fairseq

      Public
      Facebook AI Research Sequence-to-Sequence Toolkit written in Python.
      Python
      •
      Other
      •6.4k•22•0•0•Updated Jan 25, 2023Jan 25, 2023