pytorch

All

67 repositories

executorch
Public
On-device AI across mobile, embedded and edge for PyTorch
machine-learning mobile embedded
machine-learning mobile embedded deep-learning neural-network gpu tensor
Python
•
Other
•872•4.4k•865•928•Updated Mar 15, 2026Mar 15, 2026
pytorch
Public
Tensors and Dynamic neural networks in Python with strong GPU acceleration
python machine-learning deep-learning
python machine-learning deep-learning neural-network gpu numpy autograd tensor
Python
•
Other
•27k•98k•16k•2k•Updated Mar 15, 2026Mar 15, 2026
rl
Public
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
machine-learning control reinforcement-learning
machine-learning control reinforcement-learning ai robotics decision-making distributed-computing torch pytorch rl
Python
•
MIT License
•442•3.3k•199•85•Updated Mar 15, 2026Mar 15, 2026
tensordict
Public
TensorDict is a pytorch dedicated tensor container.
Python
•
MIT License
•112•1k•55•56•Updated Mar 15, 2026Mar 15, 2026
cppdocs
Public
PyTorch C++ API Documentation
HTML
•38•249•7•3•Updated Mar 15, 2026Mar 15, 2026
torchtitan
Public
A PyTorch native platform for training generative AI models
Python
•
BSD 3-Clause "New" or "Revised" License
•747•5.1k•187•226•Updated Mar 15, 2026Mar 15, 2026
vision
Public
Datasets, Transforms and Models specific to Computer Vision
machine-learning computer-vision
machine-learning computer-vision
Python
•
BSD 3-Clause "New" or "Revised" License
•7.2k•18k•914•280•Updated Mar 15, 2026Mar 15, 2026
test-infra
Public
This repository hosts code that supports the testing infrastructure for the PyTorch organization. For example, this repo hosts the logic to track disabled tests…
TypeScript
•
Other
•116•107•237•262•Updated Mar 15, 2026Mar 15, 2026
docs
Public
This repository is automatically generated to contain the website source for the PyTorch documentation at https//pytorch.org/docs.
HTML
•
BSD 3-Clause "New" or "Revised" License
•9•11•3•3•Updated Mar 15, 2026Mar 15, 2026
ci-infra
Public
Python
•
Other
•8•14•28•3•Updated Mar 14, 2026Mar 14, 2026
ao
Public
PyTorch native quantization and sparsity for training and inference
training sparsity cuda
training sparsity cuda inference pytorch transformer llama quantization mx brrr
Python
•
Other
•458•2.7k•398•280•Updated Mar 14, 2026Mar 14, 2026
helion
Public
A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
Python
•
BSD 3-Clause "New" or "Revised" License
•130•801•81•93•Updated Mar 14, 2026Mar 14, 2026
FBGEMM
Public
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
C++
•
Other
•727•1.5k•62•547•Updated Mar 14, 2026Mar 14, 2026
TensorRT
Public
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
machine-learning deep-learning cuda
machine-learning deep-learning cuda pytorch nvidia jetson tensorrt libtorch
Python
•
BSD 3-Clause "New" or "Revised" License
•384•3k•300•34•Updated Mar 14, 2026Mar 14, 2026
audio
Public
Data manipulation and transformation for audio signal processing, powered by PyTorch
audio python machine-learning
audio python machine-learning speech pytorch io audio-processing
Python
•
BSD 2-Clause "Simplified" License
•764•2.8k•240•80•Updated Mar 14, 2026Mar 14, 2026
kineto
Public
A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
HTML
•
Other
•231•931•70•44•Updated Mar 14, 2026Mar 14, 2026
benchmark
Public
TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
benchmark pytorch
benchmark pytorch
Python
•
BSD 3-Clause "New" or "Revised" License
•332•1k•82•77•Updated Mar 14, 2026Mar 14, 2026
tutorials
Public
PyTorch tutorials.
Python
•
BSD 3-Clause "New" or "Revised" License
•4.4k•9.1k•206•27•Updated Mar 13, 2026Mar 13, 2026
pytorch-integration-testing
Public
Testing downstream libraries using pytorch release candidates
Makefile
•
Other
•25•15•1•9•Updated Mar 13, 2026Mar 13, 2026
pytorch_sphinx_theme
Public
PyTorch Sphinx Theme
CSS
•
MIT License
•146•105•21•12•Updated Mar 12, 2026Mar 12, 2026
cpuinfo
Public
CPU INFOrmation library (x86/x86-64/ARM/ARM64, Linux/Windows/Android/macOS/iOS)
cpu cpuid cpu-cache
cpu cpuid cpu-cache cpu-model instruction-set cpu-topology
C
•
BSD 2-Clause "Simplified" License
•396•1.2k•47•45•Updated Mar 12, 2026Mar 12, 2026
gloo
Public
Collective communications library with various primitives for multi-machine training.
pytorch distributed-training collectives
pytorch distributed-training collectives
C++
•
Other
•352•1.4k•65•51•Updated Mar 11, 2026Mar 11, 2026
ignite
Public
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
python machine-learning deep-learning
python machine-learning deep-learning neural-network metrics pytorch hacktoberfest closember
Python
•
BSD 3-Clause "New" or "Revised" License
•686•4.8k•132•64•Updated Mar 11, 2026Mar 11, 2026
alerting-infra
Public
Alerting system for the pytorch org
TypeScript
•
BSD 3-Clause "New" or "Revised" License
•3•2•256•5•Updated Mar 2, 2026Mar 2, 2026
pytorch.github.io
Public
The website for PyTorch
HTML
•
BSD 3-Clause "New" or "Revised" License
•314•278•152•61•Updated Feb 20, 2026Feb 20, 2026
ort
Public
Accelerate PyTorch models with ONNX Runtime
Python
•
MIT License
•54•370•13•8•Updated Feb 5, 2026Feb 5, 2026
extension-cpp
Public
C++ extensions in PyTorch
Python
•252•1.2k•44•8•Updated Jan 13, 2026Jan 13, 2026
xla
Public
Enabling PyTorch on XLA Devices (e.g. Google TPU)
deep-learning compiler pytorch
deep-learning compiler pytorch xla
C++
•
Other
•566•2.8k•696•185•Updated Dec 18, 2025Dec 18, 2025
tensorpipe
Public
A tensor-aware point-to-point communication primitive for machine learning
C++
•
Other
•79•285•47•14•Updated Dec 17, 2025Dec 17, 2025
torchchat
Public archive
Run PyTorch LLMs locally on servers, desktop and mobile
local pytorch llm
local pytorch llm
Python
•
BSD 3-Clause "New" or "Revised" License
•246•3.6k•79•15•Updated Sep 10, 2025Sep 10, 2025