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

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

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

PyTorch media decoding and encoding

Testing downstream libraries using pytorch release candidates