Tensors and Dynamic neural networks in Python with strong GPU acceleration

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

Run PyTorch LLMs locally on servers, desktop and mobile

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

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 native quantization and sparsity for training and inference

Pytorch domain library for recommendation systems