Skip to content

@pytorch-tpu pytorch-tpu

Change the repository type filter

All

    Repositories list

    12 repositories

    • transformers

      Public
      🤗 Transformers: State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch.
      Python
      •
      Apache License 2.0
      •29k•17•2•4•Updated Jun 5, 2025Jun 5, 2025

    • 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 Apr 4, 2025Apr 4, 2025

    • llama

      Public
      Inference code for LLaMA models
      Python
      •
      Other
      •9.8k•21•4•7•Updated Apr 3, 2025Apr 3, 2025

    • 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
      •3•0•0•2•Updated Nov 19, 2024Nov 19, 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
      •1.1k•0•0•0•Updated Oct 17, 2024Oct 17, 2024

    • llm-foundry

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

    • composer

      Public
      Supercharge Your Model Training
      Python
      •
      Apache License 2.0
      •447•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
      •571•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.5k•22•0•0•Updated Jan 25, 2023Jan 25, 2023