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

Tensors and Dynamic neural networks in Python with strong GPU acceleration

TensorDict is a pytorch dedicated tensor container.

PyTorch native quantization and sparsity for training and inference

This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic to track disabled tests and slow tests, as well as our continuation integration jobs HUD/dashboard.

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

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

A PyTorch native platform for training generative AI models

PyTorch media decoding and encoding

PyTorch native post-training library