Meta PyTorch (meta-pytorch)

Meta PyTorch

meta-pytorch

Organization data from Github https://github.com/meta-pytorch

Home Page:https://pytorch.org

GitHub:@meta-pytorch

Meta PyTorch's repositories

torchtune

PyTorch native post-training library

Language:PythonLicense:BSD-3-ClauseStargazers:5581Issues:42Issues:1092

captum

Model interpretability and understanding for PyTorch

Language:PythonLicense:BSD-3-ClauseStargazers:5460Issues:293Issues:575

botorch

Bayesian optimization in PyTorch

Language:Jupyter NotebookLicense:MITStargazers:3396Issues:48Issues:598

torchrec

Pytorch domain library for recommendation systems

Language:PythonLicense:BSD-3-ClauseStargazers:2388Issues:35Issues:243

opacus

Training PyTorch models with differential privacy

Language:Jupyter NotebookLicense:Apache-2.0Stargazers:1878Issues:44Issues:337

tnt

A lightweight library for PyTorch training tools and utilities

Language:PythonLicense:NOASSERTIONStargazers:1715Issues:40Issues:73

data

A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.

Language:PythonLicense:BSD-3-ClauseStargazers:1232Issues:32Issues:542

attention-gym

Helpful tools and examples for working with flex-attention

Language:PythonLicense:BSD-3-ClauseStargazers:1045Issues:12Issues:109

monarch

PyTorch Single Controller

Language:RustLicense:BSD-3-ClauseStargazers:871Issues:22Issues:105

torchcodec

PyTorch media decoding and encoding

Language:PythonLicense:BSD-3-ClauseStargazers:795Issues:13Issues:233

OpenEnv

An interface library for RL post training with environments.

Language:PythonLicense:BSD-3-ClauseStargazers:668Issues:0Issues:0

torchforge

PyTorch-native post-training at scale

Language:PythonLicense:BSD-3-ClauseStargazers:509Issues:0Issues:0

torchft

Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)

Language:PythonLicense:NOASSERTIONStargazers:443Issues:18Issues:42

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:PythonLicense:NOASSERTIONStargazers:399Issues:17Issues:202

tritonbench

Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.

Language:PythonLicense:BSD-3-ClauseStargazers:280Issues:8Issues:69

torchcomms

torchcomms: a modern PyTorch communications API

Language:C++License:BSD-3-ClauseStargazers:248Issues:0Issues:0

tritonparse

TritonParse: A Compiler Tracer, Visualizer, and Reproducer for Triton Kernels

Language:PythonLicense:BSD-3-ClauseStargazers:169Issues:6Issues:36

csprng

Cryptographically secure pseudorandom number generators for PyTorch

Language:BatchfileLicense:BSD-3-ClauseStargazers:124Issues:29Issues:46

BackendBench

How to ensure correctness and ship LLM generated kernels in PyTorch

Language:PythonLicense:BSD-3-ClauseStargazers:114Issues:1Issues:57

KernelAgent

Autonomous GPU Kernel Generation via Deep Agents

Language:PythonLicense:Apache-2.0Stargazers:95Issues:0Issues:0

kraken

Triton-based Symmetric Memory operators and examples

Language:PythonLicense:NOASSERTIONStargazers:61Issues:2Issues:0

tokenizers

C++ implementations for various tokenizers (sentencepiece, tiktoken etc).

Language:C++License:BSD-3-ClauseStargazers:40Issues:9Issues:8

torchstore

A storage solution for PyTorch tensors with distributed tensor support.

Language:PythonLicense:BSD-3-ClauseStargazers:37Issues:0Issues:0

executorch-examples

Example apps and demos using PyTorch's ExecuTorch framework

Language:JavaLicense:NOASSERTIONStargazers:31Issues:12Issues:7

autoparallel

An experimental implementation of compiler-driven automatic sharding of models across a given device mesh.

Language:PythonLicense:BSD-3-ClauseStargazers:24Issues:0Issues:0

torchvision-extra-decoders

An extension of `torchvision` for decoding AVIF and HEIC images.

Language:C++License:LGPL-2.1Stargazers:9Issues:4Issues:3

chakra_replay

Chakra et_replay is a tool designed for replaying Chakra Execution Traces (ET) from machine learning models using PyTorch.

Language:PythonLicense:Apache-2.0Stargazers:7Issues:0Issues:0

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:PythonLicense:NOASSERTIONStargazers:4Issues:0Issues:2

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:PythonLicense:BSD-3-ClauseStargazers:2Issues:6Issues:0

meta-pytorch.github.io

The website for Meta-PyTorch.org

Language:HTMLLicense:BSD-3-ClauseStargazers:1Issues:0Issues:0