pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Facebook AI Research Sequence-to-Sequence Toolkit written in Python.

Datasets, Transforms and Models specific to Computer Vision

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

On-device AI across mobile, embedded and edge for PyTorch

Serve, optimize and scale PyTorch models in production

Models, data loaders and abstractions for language processing, powered by PyTorch

ELF: a platform for game research with AlphaGoZero/AlphaZero reimplementation

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

Data manipulation and transformation for audio signal processing, powered by PyTorch

Enabling PyTorch on XLA Devices (e.g. Google TPU)

Pytorch domain library for recommendation systems

Training PyTorch models with differential privacy

A lightweight library for PyTorch training tools and utilities

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.

Translate - a PyTorch Language Library

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