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 96.4k 26.4k
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
Python 23.7k 9.8k
PyTorch tutorials.
Python 9k 4.3k
Datasets, Transforms and Models specific to Computer Vision
Python 17.4k 7.2k
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
Data manipulation and transformation for audio signal processing, powered by PyTorch
A PyTorch native platform for training generative AI models
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
PyTorch C++ API Documentation
Testing downstream libraries using pytorch release candidates
PyTorch native quantization and sparsity for training and inference