Tensors and Dynamic neural networks in Python with strong GPU acceleration

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

A PyTorch native platform for training generative AI models

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

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

Datasets, Transforms and Models specific to Computer Vision

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

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.

A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.

PyTorch native quantization and sparsity for training and inference