Tensors and Dynamic neural networks in Python with strong GPU acceleration

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

PyTorch C++ API Documentation

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

This repository is automatically generated to contain the website source for the PyTorch documentation at https//pytorch.org/docs.

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

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.

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

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.