PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

Tensors and Dynamic neural networks in Python with strong GPU acceleration

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

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

A PyTorch native platform for training generative AI models

A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.

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

FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) -https://code.fb.com/ml-applications/fbgemm/

Datasets, Transforms and Models specific to Computer Vision

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