This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our continuation integration jobs HUD/dashboard.

A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.

Alerting system for the pytorch org

Collective communications library with various primitives for multi-machine training.

A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.

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