A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
PyTorch media decoding and encoding
Alerting system for the pytorch org
Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)
Collective communications library with various primitives for multi-machine training.
torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters in a single...
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
Pytorch domain library for recommendation systems
A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
Example repository for custom C++/CUDA operators for TorchScript
ELF: a platform for game research with AlphaGoZero/AlphaZero reimplementation
C++ extensions in PyTorch
Datasets, Transforms and Models specific to Computer Vision
Models, data loaders and abstractions for language processing, powered by PyTorch
