We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see ourdocumentation.
Tensors and Dynamic neural networks in Python with strong GPU acceleration
Python 96k 26.3k
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
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) -https://code.fb.com/ml-applications/fbgemm/
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
On-device AI across mobile, embedded and edge for PyTorch
A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
Data manipulation and transformation for audio signal processing, powered by PyTorch
Testing downstream libraries using pytorch release candidates