Skip to content

@meta-pytorch Meta PyTorch

Change the repository type filter

All

    Repositories list

    51 repositories

    • tritonparse

      Public
      TritonParse: A Compiler Tracer, Visualizer, and Reproducer for Triton Kernels
      debuggingcompilerpytorchtritonstructured-logginginteractive-visualizationir-analysisgpu-kernelir-visualization
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •15•179•43•9•Updated Dec 27, 2025Dec 27, 2025

    • monarch

      Public
      PyTorch Single Controller
      Rust
      •
      BSD 3-Clause "New" or "Revised" License
      •121•933•131•276•Updated Dec 27, 2025Dec 27, 2025

    • torchcomms

      Public
      torchcomms: a modern PyTorch communications API
      C++
      •
      BSD 3-Clause "New" or "Revised" License
      •50•314•21•28•Updated Dec 27, 2025Dec 27, 2025

    • captum

      Public
      Model interpretability and understanding for PyTorch
      interpretabilityfeature-importanceinterpretable-aiinterpretable-mlfeature-attribution
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •549•5.5k•222•55•Updated Dec 26, 2025Dec 26, 2025

    • 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
      •
      Other
      •9•15•0•5•Updated Dec 26, 2025Dec 26, 2025

    • tlparse

      Public
      TORCH_LOGS parser for PT2
      Rust
      •
      BSD 3-Clause "New" or "Revised" License
      •22•70•37•4•Updated Dec 26, 2025Dec 26, 2025

    • compile-graph-break-site

      Public
      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
      •
      Other
      •3•4•4•0•Updated Dec 26, 2025Dec 26, 2025

    • torchft

      Public
      Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
      trainingdistributed-systemsfault-toleranceconsensusmltorchpytorchhsdpfsdplocalsgd
      Python
      •
      Other
      •53•462•30•24•Updated Dec 26, 2025Dec 26, 2025

    • torchforge

      Public
      PyTorch-native post-training at scale
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •72•576•51•46•Updated Dec 26, 2025Dec 26, 2025

    • torchvision-extra-decoders

      Public
      An extension of `torchvision` for decoding AVIF and HEIC images.
      C++
      •
      GNU Lesser General Public License v2.1
      •2•9•2•1•Updated Dec 26, 2025Dec 26, 2025

    • torchcodec

      Public
      PyTorch media decoding and encoding
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •80•884•83•27•Updated Dec 26, 2025Dec 26, 2025

    • torchtune

      Public
      PyTorch native post-training library
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •693•5.6k•341•101•Updated Dec 26, 2025Dec 26, 2025

    • torchrec

      Public
      Pytorch domain library for recommendation systems
      deep-learninggpucudapytorchrecommendation-systemshardingrecommender-system
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •586•2.4k•85•97•Updated Dec 26, 2025Dec 26, 2025

    • data

      Public
      A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •170•1.2k•223•32•Updated Dec 26, 2025Dec 26, 2025

    • tnt

      Public
      A lightweight library for PyTorch training tools and utilities
      pythonmachine-learningreinforcement-learningdeep-learningneural-networkpytorch
      Python
      •
      Other
      •297•1.7k•7•91•Updated Dec 24, 2025Dec 24, 2025

    • OpenEnv

      Public
      An interface library for RL post training with environments.
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •138•861•26•28•Updated Dec 24, 2025Dec 24, 2025

    • torch-release-notes

      Public
      Staging ground for pytorch/pytorch maintainers to collaborate on release notes for PyTorch releases. Houses the artifacts that chronicle each release.
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •26•2•0•2•Updated Dec 24, 2025Dec 24, 2025

    • 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.
      pythonkubernetescomponentsmachine-learningairflowdeep-learningslurmpipelinespytorchray
      Python
      •
      Other
      •148•408•67•51•Updated Dec 23, 2025Dec 23, 2025

    • tritonbench

      Public
      Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •55•306•16•33•Updated Dec 23, 2025Dec 23, 2025

    • autoparallel

      Public
      An experimental implementation of compiler-driven automatic sharding of models across a given device mesh.
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •13•48•7•24•Updated Dec 23, 2025Dec 23, 2025

    • torchsnapshot

      Public
      A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind.
      Python
      •
      Other
      •43•162•10•10•Updated Dec 22, 2025Dec 22, 2025

    • attention-gym

      Public
      Helpful tools and examples for working with flex-attention
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •67•1.1k•68•3•Updated Dec 22, 2025Dec 22, 2025

    • botorch

      Public
      Bayesian optimization in PyTorch
      Jupyter Notebook
      •
      MIT License
      •457•3.4k•67•25•Updated Dec 22, 2025Dec 22, 2025

    • KernelAgent

      Public
      Autonomous GPU Kernel Generation via Deep Agents
      Python
      •
      Apache License 2.0
      •21•192•5•3•Updated Dec 20, 2025Dec 20, 2025

    • torcheval

      Public
      A library that contains a rich collection of performant PyTorch model metrics, a simple interface to create new metrics, a toolkit to facilitate metric computation in distributed training and tools for PyTorch model evaluations.
      Python
      •
      Other
      •56•245•20•12•Updated Dec 20, 2025Dec 20, 2025

    • executorch-examples

      Public
      Example apps and demos using PyTorch's ExecuTorch framework
      Java
      •
      Other
      •17•49•8•9•Updated Dec 19, 2025Dec 19, 2025

    • torchstore

      Public
      A storage solution for PyTorch tensors with distributed tensor support.
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •6•48•10•5•Updated Dec 19, 2025Dec 19, 2025

    • BackendBench

      Public
      Ship correct and fast LLM kernels to PyTorch
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •15•127•18•20•Updated Dec 18, 2025Dec 18, 2025

    • tokenizers

      Public
      C++ implementations for various tokenizers (sentencepiece, tiktoken etc).
      C++
      •
      BSD 3-Clause "New" or "Revised" License
      •22•44•5•13•Updated Dec 16, 2025Dec 16, 2025

    • multipy

      Public
      torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters in a single C++ process.
      C++
      •
      Other
      •36•182•37•20•Updated Dec 16, 2025Dec 16, 2025