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 97.9k 27k
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
Python 23.8k 9.8k
PyTorch tutorials.
Python 9.1k 4.4k
Datasets, Transforms and Models specific to Computer Vision
Python 17.5k 7.2k
On-device AI across mobile, embedded and edge for PyTorch
A PyTorch native platform for training generative AI models
TensorDict is a pytorch dedicated tensor container.
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
PyTorch C++ API Documentation
A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
