pytorch

All

67 repositories

pytorch
Public
Tensors and Dynamic neural networks in Python with strong GPU acceleration
python machine-learning deep-learning neural-network gpu numpy autograd tensor
Python
•
Other
•27k•98k•16k•1.9k•Updated Feb 20, 2026Feb 20, 2026
ao
Public
PyTorch native quantization and sparsity for training and inference
training sparsity cuda inference pytorch transformer llama quantization mx brrr
Python
•
Other
•432•2.7k•385•245•Updated Feb 20, 2026Feb 20, 2026
executorch
Public
On-device AI across mobile, embedded and edge for PyTorch
machine-learning mobile embedded deep-learning neural-network gpu tensor
Python
•
Other
•847•4.3k•836•796•Updated Feb 20, 2026Feb 20, 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
•104•761•83•44•Updated Feb 20, 2026Feb 20, 2026
TensorRT
Public
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
machine-learning deep-learning cuda pytorch nvidia jetson tensorrt libtorch
Python
•
BSD 3-Clause "New" or "Revised" License
•382•2.9k•297•41•Updated Feb 20, 2026Feb 20, 2026
torchtitan
Public
A PyTorch native platform for training generative AI models
Python
•
BSD 3-Clause "New" or "Revised" License
•709•5.1k•180•207•Updated Feb 20, 2026Feb 20, 2026
FBGEMM
Public
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
C++
•
Other
•721•1.5k•62•612•Updated Feb 20, 2026Feb 20, 2026
vision
Public
Datasets, Transforms and Models specific to Computer Vision
machine-learning computer-vision
Python
•
BSD 3-Clause "New" or "Revised" License
•7.2k•18k•913•272•Updated Feb 19, 2026Feb 19, 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
•112•106•236•252•Updated Feb 19, 2026Feb 19, 2026
tutorials
Public
PyTorch tutorials.
Python
•
BSD 3-Clause "New" or "Revised" License
•4.4k•9k•204•45•Updated Feb 19, 2026Feb 19, 2026
kineto
Public
A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
HTML
•
Other
•226•923•69•33•Updated Feb 19, 2026Feb 19, 2026
ignite
Public
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
python machine-learning deep-learning neural-network metrics pytorch hacktoberfest closember
Python
•
BSD 3-Clause "New" or "Revised" License
•671•4.7k•116•55•Updated Feb 19, 2026Feb 19, 2026
tensordict
Public
TensorDict is a pytorch dedicated tensor container.
Python
•
MIT License
•112•1k•52•44•Updated Feb 19, 2026Feb 19, 2026
rl
Public
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
machine-learning control reinforcement-learning ai robotics decision-making distributed-computing torch pytorch rl
Python
•
MIT License
•438•3.3k•203•86•Updated Feb 19, 2026Feb 19, 2026
pytorch_sphinx_theme
Public
PyTorch Sphinx Theme
CSS
•
MIT License
•145•105•21•10•Updated Feb 19, 2026Feb 19, 2026
audio
Public
Data manipulation and transformation for audio signal processing, powered by PyTorch
audio python machine-learning speech pytorch io audio-processing
Python
•
BSD 2-Clause "Simplified" License
•764•2.8k•238•81•Updated Feb 19, 2026Feb 19, 2026
cppdocs
Public
PyTorch C++ API Documentation
HTML
•38•246•7•3•Updated Feb 19, 2026Feb 19, 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•10•3•0•Updated Feb 19, 2026Feb 19, 2026
cpuinfo
Public
CPU INFOrmation library (x86/x86-64/ARM/ARM64, Linux/Windows/Android/macOS/iOS)
cpu cpuid cpu-cache cpu-model instruction-set cpu-topology
C
•
BSD 2-Clause "Simplified" License
•389•1.2k•48•45•Updated Feb 18, 2026Feb 18, 2026
benchmark
Public
TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
benchmark pytorch
Python
•
BSD 3-Clause "New" or "Revised" License
•329•1k•81•73•Updated Feb 18, 2026Feb 18, 2026
pytorch.github.io
Public
The website for PyTorch
HTML
•
BSD 3-Clause "New" or "Revised" License
•312•276•152•62•Updated Feb 17, 2026Feb 17, 2026
pytorch-integration-testing
Public
Testing downstream libraries using pytorch release candidates
Makefile
•
Other
•24•15•1•5•Updated Feb 16, 2026Feb 16, 2026
gloo
Public
Collective communications library with various primitives for multi-machine training.
pytorch distributed-training collectives
C++
•
Other
•347•1.4k•65•45•Updated Feb 12, 2026Feb 12, 2026
alerting-infra
Public
Alerting system for the pytorch org
TypeScript
•
BSD 3-Clause "New" or "Revised" License
•3•2•191•2•Updated Feb 11, 2026Feb 11, 2026
ci-infra
Public
HCL
•
Other
•6•13•28•0•Updated Feb 9, 2026Feb 9, 2026
ort
Public
Accelerate PyTorch models with ONNX Runtime
Python
•
MIT License
•55•368•13•8•Updated Feb 5, 2026Feb 5, 2026
extension-cpp
Public
C++ extensions in PyTorch
Python
•251•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 xla
C++
•
Other
•564•2.7k•694•185•Updated Dec 18, 2025Dec 18, 2025
tensorpipe
Public
A tensor-aware point-to-point communication primitive for machine learning
C++
•
Other
•79•284•47•14•Updated Dec 17, 2025Dec 17, 2025
torchchat
Public archive
Run PyTorch LLMs locally on servers, desktop and mobile
local pytorch llm
Python
•
BSD 3-Clause "New" or "Revised" License
•248•3.6k•79•15•Updated Sep 10, 2025Sep 10, 2025