Skip to content

@meta-pytorch Meta PyTorch

Change the repository type filter

All

    Repositories list

    51 repositories

    • 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
      •54•302•17•33•Updated Dec 15, 2025Dec 15, 2025

    • monarch

      Public
      PyTorch Single Controller
      Rust
      •
      BSD 3-Clause "New" or "Revised" License
      •117•921•138•261•Updated Dec 15, 2025Dec 15, 2025

    • torchcomms

      Public
      torchcomms: a modern PyTorch communications API
      C++
      •
      BSD 3-Clause "New" or "Revised" License
      •43•308•20•20•Updated Dec 15, 2025Dec 15, 2025

    • torchforge

      Public
      PyTorch-native post-training at scale
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •71•567•45•38•Updated Dec 14, 2025Dec 14, 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 14, 2025Dec 14, 2025

    • torchcodec

      Public
      PyTorch media decoding and encoding
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •77•863•77•28•Updated Dec 14, 2025Dec 14, 2025

    • torchtune

      Public
      PyTorch native post-training library
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •689•5.6k•341•99•Updated Dec 14, 2025Dec 14, 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
      •3•9•0•6•Updated Dec 14, 2025Dec 14, 2025

    • torchrec

      Public
      Pytorch domain library for recommendation systems
      deep-learninggpucudapytorchrecommendation-systemshardingrecommender-system
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •584•2.4k•85•99•Updated Dec 14, 2025Dec 14, 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 14, 2025Dec 14, 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
      •2•4•4•0•Updated Dec 13, 2025Dec 13, 2025

    • torchstore

      Public
      A storage solution for PyTorch tensors with distributed tensor support.
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •6•47•8•7•Updated Dec 13, 2025Dec 13, 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
      •12•47•7•18•Updated Dec 13, 2025Dec 13, 2025

    • KernelAgent

      Public
      Autonomous GPU Kernel Generation via Deep Agents
      Python
      •
      Apache License 2.0
      •20•181•5•7•Updated Dec 13, 2025Dec 13, 2025

    • 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•178•39•0•Updated Dec 13, 2025Dec 13, 2025

    • executorch-examples

      Public
      Example apps and demos using PyTorch's ExecuTorch framework
      Java
      •
      Other
      •17•42•7•6•Updated Dec 11, 2025Dec 11, 2025

    • OpenEnv

      Public
      An interface library for RL post training with environments.
      Python
      •
      BSD 3-Clause "New" or "Revised" License
      •128•842•22•30•Updated Dec 11, 2025Dec 11, 2025

    • botorch

      Public
      Bayesian optimization in PyTorch
      Jupyter Notebook
      •
      MIT License
      •458•3.4k•66•27•Updated Dec 11, 2025Dec 11, 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 10, 2025Dec 10, 2025

    • tokenizers

      Public
      C++ implementations for various tokenizers (sentencepiece, tiktoken etc).
      C++
      •
      BSD 3-Clause "New" or "Revised" License
      •20•43•5•16•Updated Dec 10, 2025Dec 10, 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•244•20•12•Updated Dec 10, 2025Dec 10, 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
      •147•406•67•51•Updated Dec 9, 2025Dec 9, 2025

    • torchft

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

    • attention-gym

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

    • captum

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

    • tnt

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

    • chakra_replay

      Public
      Chakra et_replay is a tool designed for replaying Chakra Execution Traces (ET) from machine learning models using PyTorch.
      Python
      •
      Apache License 2.0
      •3•7•1•4•Updated Nov 21, 2025Nov 21, 2025

    • opacus

      Public
      Training PyTorch models with differential privacy
      machine-learningdeep-learningneural-networkpytorchdifferential-privacyprivacy-preserving-machine-learning
      Jupyter Notebook
      •
      Apache License 2.0
      •387•1.9k•62•9•Updated Nov 12, 2025Nov 12, 2025

    • csprng

      Public
      Cryptographically secure pseudorandom number generators for PyTorch
      Batchfile
      •
      BSD 3-Clause "New" or "Revised" License
      •30•125•22•7•Updated Nov 12, 2025Nov 12, 2025

    • tlparse

      Public
      TORCH_LOGS parser for PT2
      Rust
      •
      BSD 3-Clause "New" or "Revised" License
      •22•69•38•4•Updated Nov 10, 2025Nov 10, 2025