pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Enabling PyTorch on XLA Devices (e.g. Google TPU)

PyTorch native quantization and sparsity for training and inference

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

On-device AI across mobile, embedded and edge for 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/dasard.

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

A PyTorch native platform for training generative AI models

Model interpretability and understanding for PyTorch

Bayesian optimization in PyTorch