We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Python 95.9k 26.2k
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
Python 23.6k 9.8k
PyTorch tutorials.
Python 8.9k 4.3k
Datasets, Transforms and Models specific to Computer Vision
Python 17.4k 7.2k
PyTorch native quantization and sparsity for training and inference
On-device AI across mobile, embedded and edge for PyTorch
A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
This repository is automatically generated to contain the website source for the PyTorch documentation at https//pytorch.org/docs.
TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
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
