Skip to content
Change the repository type filter

All

    Repositories list

    • torchcomms: a modern PyTorch communications API
      C++
      7032224141Updated Jan 25, 2026Jan 25, 2026
    • torchx

      Public
      TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and support for E2E production ML pipelines when you're ready.
      Python
      1484116756Updated Jan 25, 2026Jan 25, 2026
    • PyTorch-native post-training at scale
      Python
      806005051Updated Jan 24, 2026Jan 24, 2026
    • torchvision-extra-decoders

      Public
      An extension of `torchvision` for decoding AVIF and HEIC images.
      C++
      2921Updated Jan 24, 2026Jan 24, 2026
    • torchcodec

      Public
      PyTorch media decoding and encoding
      Python
      879298833Updated Jan 24, 2026Jan 24, 2026
    • torchtune

      Public
      PyTorch native post-training library
      Python
      6985.7k343103Updated Jan 24, 2026Jan 24, 2026
    • MSLK

      Public
      MSLK (Meta Superintelligence Labs Kernels) is a collection of PyTorch GPU operator libraries that are designed and optimized for GenAI training and inference, such as FP8 row-wise quantization and collective communications.
      Python
      163705Updated Jan 24, 2026Jan 24, 2026
    • torchrec

      Public
      Pytorch domain library for recommendation systems
      Python
      5962.5k88101Updated Jan 24, 2026Jan 24, 2026
    • data

      Public
      A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
      Python
      1701.2k22133Updated Jan 24, 2026Jan 24, 2026
    • KernelAgent

      Public
      Autonomous GPU Kernel Generation via Deep Agents
      Python
      27220512Updated Jan 24, 2026Jan 24, 2026
    • captum

      Public
      Model interpretability and understanding for PyTorch
      Python
      5505.5k22331Updated Jan 24, 2026Jan 24, 2026
    • Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
      Python
      613181829Updated Jan 24, 2026Jan 24, 2026
    • monarch

      Public
      PyTorch Single Controller
      Rust
      127953140299Updated Jan 24, 2026Jan 24, 2026
    • Example apps and demos using PyTorch's ExecuTorch framework
      Kotlin
      215976Updated Jan 23, 2026Jan 23, 2026
    • A storage solution for PyTorch tensors with distributed tensor support.
      Python
      7601011Updated Jan 23, 2026Jan 23, 2026
    • botorch

      Public
      Bayesian optimization in PyTorch
      Jupyter Notebook
      4613.5k6628Updated Jan 23, 2026Jan 23, 2026
    • Monarch CRD and operator for native kubernetes support
      Go
      2027Updated Jan 23, 2026Jan 23, 2026
    • An experimental implementation of compiler-driven automatic sharding of models across a given device mesh.
      Python
      1451718Updated Jan 23, 2026Jan 23, 2026
    • TritonParse: A Compiler Tracer, Visualizer, and Reproducer for Triton Kernels
      Python
      16188441Updated Jan 23, 2026Jan 23, 2026
    • OpenEnv

      Public
      An interface library for RL post training with environments.
      Python
      1701.1k2331Updated Jan 22, 2026Jan 22, 2026
    • C++ implementations for various tokenizers (sentencepiece, tiktoken etc).
      C++
      2447514Updated Jan 22, 2026Jan 22, 2026
    • This repository contains the source code for a static website that provides documentation for each "Graph Break" identified by a Graph Break ID (GBID).
      Python
      4450Updated Jan 22, 2026Jan 22, 2026
    • Chakra et_replay is a tool designed for replaying Chakra Execution Traces (ET) from machine learning models using PyTorch.
      Python
      3914Updated Jan 20, 2026Jan 20, 2026
    • Staging ground for pytorch/pytorch maintainers to collaborate on release notes for PyTorch releases. Houses the artifacts that chronicle each release.
      Python
      26200Updated Jan 20, 2026Jan 20, 2026
    • tlparse

      Public
      TORCH_TRACE parser for PT2
      Rust
      2372396Updated Jan 19, 2026Jan 19, 2026
    • Helpful tools and examples for working with flex-attention
      Python
      701.1k683Updated Jan 16, 2026Jan 16, 2026
    • kraken

      Public
      Triton-based Symmetric Memory operators and examples
      Python
      128005Updated Jan 15, 2026Jan 15, 2026
    • Ship correct and fast LLM kernels to PyTorch
      Python
      151361820Updated Jan 14, 2026Jan 14, 2026
    • A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind.
      Python
      431641010Updated Jan 12, 2026Jan 12, 2026
    • tnt

      Public
      A lightweight library for PyTorch training tools and utilities
      Python
      2951.7k790Updated Jan 12, 2026Jan 12, 2026