pytorch

All

67 repositories

pytorch
Public
Tensors and Dynamic neural networks in Python with strong GPU acceleration
machine-learning deep-learning python neural-network gpu numpy autograd tensor
Python
•
Other
•26k•94k•15k•1.5k•Updated Oct 13, 2025Oct 13, 2025
executorch
Public
On-device AI across mobile, embedded and edge for PyTorch
machine-learning mobile embedded deep-learning neural-network gpu tensor
Python
•
Other
•685•3.3k•766•692•Updated Oct 13, 2025Oct 13, 2025
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
•38•341•56•35•Updated Oct 13, 2025Oct 13, 2025
TensorRT
Public
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
machine-learning nvidia jetson tensorrt libtorch deep-learning cuda pytorch
Python
•
BSD 3-Clause "New" or "Revised" License
•369•2.9k•247•26•Updated Oct 13, 2025Oct 13, 2025
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 and slow tests, as well as our continuation integration jobs HUD/dashboard.
TypeScript
•
Other
•105•102•215•236•Updated Oct 13, 2025Oct 13, 2025
tutorials
Public
PyTorch tutorials.
Python
•
BSD 3-Clause "New" or "Revised" License
•4.3k•8.8k•170•23•Updated Oct 13, 2025Oct 13, 2025
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
•5•9•2•2•Updated Oct 13, 2025Oct 13, 2025
ao
Public
PyTorch native quantization and sparsity for training and inference
training sparsity cuda inference optimizer pytorch transformer offloading llama quantization
Python
•
Other
•345•2.4k•323•191•Updated Oct 13, 2025Oct 13, 2025
FBGEMM
Public
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
C++
•
Other
•659•1.5k•54•546•Updated Oct 13, 2025Oct 13, 2025
torchtitan
Public
A PyTorch native platform for training generative AI models
Python
•
BSD 3-Clause "New" or "Revised" License
•560•4.5k•121•141•Updated Oct 13, 2025Oct 13, 2025
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
•323•987•80•78•Updated Oct 13, 2025Oct 13, 2025
vision
Public
Datasets, Transforms and Models specific to Computer Vision
machine-learning computer-vision
Python
•
BSD 3-Clause "New" or "Revised" License
•7.2k•17k•894•243•Updated Oct 13, 2025Oct 13, 2025
audio
Public
Data manipulation and transformation for audio signal processing, powered by PyTorch
python machine-learning speech io audio pytorch audio-processing
Python
•
BSD 2-Clause "Simplified" License
•728•2.8k•241•93•Updated Oct 13, 2025Oct 13, 2025
cppdocs
Public
PyTorch C++ API Documentation
HTML
•37•240•7•3•Updated Oct 13, 2025Oct 13, 2025
kineto
Public
A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
HTML
•
Other
•207•881•63•23•Updated Oct 11, 2025Oct 11, 2025
xla
Public
Enabling PyTorch on XLA Devices (e.g. Google TPU)
compiler xla deep-learning pytorch
C++
•
Other
•559•2.7k•691•189•Updated Oct 10, 2025Oct 10, 2025
pytorch-integration-testing
Public
Testing downstream libraries using pytorch release candidates
Makefile
•
MIT License
•20•12•1•6•Updated Oct 8, 2025Oct 8, 2025
tensordict
Public
TensorDict is a pytorch dedicated tensor container.
Python
•
MIT License
•101•971•44•34•Updated Oct 7, 2025Oct 7, 2025
alerting-infra
Public
Alerting system for the pytorch org
TypeScript
•
BSD 3-Clause "New" or "Revised" License
•1•2•10•0•Updated Oct 6, 2025Oct 6, 2025
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
•413•3.1k•205•84•Updated Oct 2, 2025Oct 2, 2025
pytorch_sphinx_theme
Public
PyTorch Sphinx Theme
CSS
•
MIT License
•145•103•21•8•Updated Oct 2, 2025Oct 2, 2025
pytorch.github.io
Public
The website for PyTorch
HTML
•
BSD 3-Clause "New" or "Revised" License
•312•264•148•64•Updated Sep 30, 2025Sep 30, 2025
ignite
Public
High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.
neural-network metrics hacktoberfest closember python machine-learning deep-learning pytorch
Python
•
BSD 3-Clause "New" or "Revised" License
•655•4.7k•115•51•Updated Sep 25, 2025Sep 25, 2025
ci-infra
Public
HCL
•
Other
•4•14•27•0•Updated Sep 23, 2025Sep 23, 2025
gloo
Public
Collective communications library with various primitives for multi-machine training.
pytorch distributed-training collectives
C++
•
Other
•338•1.4k•62•43•Updated Sep 12, 2025Sep 12, 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
text
Public archive
Models, data loaders and abstractions for language processing, powered by PyTorch
nlp deep-learning models pytorch dataset data-loader
Python
•
BSD 3-Clause "New" or "Revised" License
•812•3.6k•290•56•Updated Sep 10, 2025Sep 10, 2025
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
•372•1.1k•50•48•Updated Sep 5, 2025Sep 5, 2025
examples
Public
A set of examples around pytorch in Vision, Text, Reinforcement Learning, etc.
Python
•
BSD 3-Clause "New" or "Revised" License
•9.8k•23k•197•43•Updated Sep 1, 2025Sep 1, 2025
functorch
Public archive
functorch is JAX-like composable function transforms for PyTorch.
gradients hessians pytorch
Jupyter Notebook
•
BSD 3-Clause "New" or "Revised" License
•105•1.4k•233•17•Updated Aug 21, 2025Aug 21, 2025