Skip to content

@pytorch-tpu pytorch-tpu

All

    Repositories list

    13 repositories

    • jax-torch-interop

      Public
      jax-torch-interop
      BSD 3-Clause "New" or "Revised" License
      •0•1•0•0•Updated Sep 18, 2025Sep 18, 2025

    • transformers

      Public
      🤗 Transformers: State-of-the-art Natural Language Processing for TensorFlow 2.0 and PyTorch.
      Python
      •
      Apache License 2.0
      •35k•17•3•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•3•0•1•Updated Apr 4, 2025Apr 4, 2025

    • llama

      Public
      Inference code for LLaMA models
      Python
      •
      Other
      •9.8k•20•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-…
      Python
      •
      Apache License 2.0
      •1.6k•0•0•0•Updated Oct 17, 2024Oct 17, 2024

    • llm-foundry

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

    • composer

      Public
      Supercharge Your Model Training
      Python
      •
      Apache License 2.0
      •473•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
      •11k•7•0•0•Updated Sep 11, 2023Sep 11, 2023

    • training

      Public
      Reference implementations of MLPerf™ training benchmarks
      Python
      •
      Other
      •597•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.7k•22•0•0•Updated Jan 25, 2023Jan 25, 2023
    ProTip! When viewing an organization's repositories, you can use the props. filter to filter by custom property.