Meta PyTorch

Pinned Repositories

attention-gym
Helpful tools and examples for working with flex-attention
Language:Python1k 12 10963
botorch
Bayesian optimization in PyTorch
Language:Jupyter Notebook3.4k 48 598452
captum
Model interpretability and understanding for PyTorch
Language:Python5.5k 293 575547
data
A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
Language:Python1.2k 32 542169
gpt-fast
Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.
Language:Python6.1k 58 120564
opacus
Training PyTorch models with differential privacy
Language:Jupyter Notebook1.9k 44 337386
segment-anything-fast
A batched offline inference oriented version of segment-anything
Language:Python1.3k 10 4674
tnt
A lightweight library for PyTorch training tools and utilities
Language:Python1.7k 40 73296
torchrec
Pytorch domain library for recommendation systems
Language:Python2.4k 35 243573
torchtune
PyTorch native post-training library
Language:Python5.6k 42 1.1k679

Meta PyTorch's Repositories

meta-pytorch/torchtune
PyTorch native post-training library
Language:Python5.6k 42 1.1k679
meta-pytorch/captum
Model interpretability and understanding for PyTorch
Language:Python5.5k 293 575547
meta-pytorch/botorch
Bayesian optimization in PyTorch
Language:Jupyter Notebook3.4k 48 598452
meta-pytorch/torchrec
Pytorch domain library for recommendation systems
Language:Python2.4k 35 243573
meta-pytorch/opacus
Training PyTorch models with differential privacy
Language:Jupyter Notebook1.9k 44 337386
meta-pytorch/tnt
A lightweight library for PyTorch training tools and utilities
Language:Python1.7k 40 73296
meta-pytorch/data
A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
Language:Python1.2k 32 542169
meta-pytorch/attention-gym
Helpful tools and examples for working with flex-attention
Language:Python1k 12 10963
meta-pytorch/monarch
PyTorch Single Controller
Language:Rust871 22 105103
meta-pytorch/torchcodec
PyTorch media decoding and encoding
Language:Python795 13 23367
meta-pytorch/OpenEnv
An interface library for RL post training with environments.
Language:Python66888
meta-pytorch/torchforge
PyTorch-native post-training at scale
Language:Python50951
meta-pytorch/torchft
Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
Language:Python443 18 4251
meta-pytorch/torchx
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.
Language:Python399 17 202147
meta-pytorch/tritonbench
Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
Language:Python280 8 6950
meta-pytorch/torchcomms
torchcomms: a modern PyTorch communications API
Language:C++24828
meta-pytorch/tritonparse
TritonParse: A Compiler Tracer, Visualizer, and Reproducer for Triton Kernels
Language:Python169 6 3613
meta-pytorch/csprng
Cryptographically secure pseudorandom number generators for PyTorch
Language:Batchfile124 29 4631
meta-pytorch/BackendBench
How to ensure correctness and ship LLM generated kernels in PyTorch
Language:Python114 1 5714
meta-pytorch/KernelAgent
Autonomous GPU Kernel Generation via Deep Agents
Language:Python959
meta-pytorch/kraken
Triton-based Symmetric Memory operators and examples
Language:Python61 2 010
meta-pytorch/tokenizers
C++ implementations for various tokenizers (sentencepiece, tiktoken etc).
Language:C++40 9 819
meta-pytorch/torchstore
A storage solution for PyTorch tensors with distributed tensor support.
Language:Python375
meta-pytorch/executorch-examples
Example apps and demos using PyTorch's ExecuTorch framework
Language:Java31 12 714
meta-pytorch/autoparallel
An experimental implementation of compiler-driven automatic sharding of models across a given device mesh.
Language:Python247
meta-pytorch/torchvision-extra-decoders
An extension of `torchvision` for decoding AVIF and HEIC images.
Language:C++9 4 32
meta-pytorch/chakra_replay
Chakra et_replay is a tool designed for replaying Chakra Execution Traces (ET) from machine learning models using PyTorch.
Language:Python73
meta-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).
Language:Python4 0 2
meta-pytorch/torch-release-notes
Staging ground for pytorch/pytorch maintainers to collaborate on release notes for PyTorch releases. Houses the artifacts that chronicle each release.
Language:Python2 6 025
meta-pytorch/meta-pytorch.github.io
The website for Meta-PyTorch.org
Language:HTML1 0 02