Meta PyTorch's repositories
attention-gym
Helpful tools and examples for working with flex-attention
torchcodec
PyTorch media decoding and encoding
torchforge
PyTorch-native post-training at scale
tritonbench
Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
torchcomms
torchcomms: a modern PyTorch communications API
tritonparse
TritonParse: A Compiler Tracer, Visualizer, and Reproducer for Triton Kernels
BackendBench
How to ensure correctness and ship LLM generated kernels in PyTorch
KernelAgent
Autonomous GPU Kernel Generation via Deep Agents
tokenizers
C++ implementations for various tokenizers (sentencepiece, tiktoken etc).
torchstore
A storage solution for PyTorch tensors with distributed tensor support.
executorch-examples
Example apps and demos using PyTorch's ExecuTorch framework
autoparallel
An experimental implementation of compiler-driven automatic sharding of models across a given device mesh.
torchvision-extra-decoders
An extension of `torchvision` for decoding AVIF and HEIC images.
chakra_replay
Chakra et_replay is a tool designed for replaying Chakra Execution Traces (ET) from machine learning models using PyTorch.
compile-graph-break-site
This repository contains the source code for a static website that provides documentation for each "Graph Break" identified by a Graph Break ID (GBID).
torch-release-notes
Staging ground for pytorch/pytorch maintainers to collaborate on release notes for PyTorch releases. Houses the artifacts that chronicle each release.
meta-pytorch.github.io
The website for Meta-PyTorch.org