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.5k 26.5k
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
TensorDict is a pytorch dedicated tensor container.
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
On-device AI across mobile, embedded and edge for PyTorch
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
PyTorch C++ API Documentation
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.
PyTorch native quantization and sparsity for training and inference
