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.

Run PyTorch LLMs locally on servers, desktop and mobile

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

On-device AI across mobile, embedded and edge for 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 domain library for recommendation systems

PyTorch native quantization and sparsity for training and inference