pytorch

All

82 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
•24k•90k•15k•1.3k•Updated May 28, 2025May 28, 2025
torchtitan
Public
A PyTorch native platform for training generative AI models
Python
•
BSD 3-Clause "New" or "Revised" License
•379•3.8k•101•63•Updated May 28, 2025May 28, 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
•97•92•228•197•Updated May 28, 2025May 28, 2025
xla
Public
Enabling PyTorch on XLA Devices (e.g. Google TPU)
compiler xla deep-learning pytorch
Python
•
Other
•527•2.6k•641•194•Updated May 28, 2025May 28, 2025
torchrec
Public
Pytorch domain library for recommendation systems
deep-learning gpu cuda pytorch recommendation-system sharding recommender-system
Python
•
BSD 3-Clause "New" or "Revised" License
•518•2.2k•80•341•Updated May 28, 2025May 28, 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
•553•2.9k•493•556•Updated May 28, 2025May 28, 2025
ao
Public
PyTorch native quantization and sparsity for training and inference
training sparsity cuda inference optimizer pytorch transformer offloading llama quantization
Python
•
BSD 3-Clause "New" or "Revised" License
•271•2.1k•245•129•Updated May 28, 2025May 28, 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
•364•2.8k•210•21•Updated May 28, 2025May 28, 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
•3•9•2•0•Updated May 28, 2025May 28, 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
•356•1.1k•43•37•Updated May 28, 2025May 28, 2025
functorch
Public
functorch is JAX-like composable function transforms for PyTorch.
gradients hessians pytorch
Jupyter Notebook
•
BSD 3-Clause "New" or "Revised" License
•104•1.4k•232•17•Updated May 28, 2025May 28, 2025
botorch
Public
Bayesian optimization in PyTorch
Jupyter Notebook
•
MIT License
•423•3.3k•60•20•Updated May 28, 2025May 28, 2025
FBGEMM
Public
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
C++
•
Other
•584•1.3k•50•460•Updated May 28, 2025May 28, 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
•308•945•79•74•Updated May 27, 2025May 27, 2025
data
Public
A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.
Python
•
BSD 3-Clause "New" or "Revised" License
•166•1.2k•219•33•Updated May 27, 2025May 27, 2025
torchtune
Public
PyTorch native post-training library
Python
•
BSD 3-Clause "New" or "Revised" License
•614•5.2k•309•88•Updated May 27, 2025May 27, 2025
torchft
Public
PyTorch per step fault tolerance (actively under development)
training distributed-systems fault-tolerance consensus ml torch pytorch
Python
•
Other
•33•302•21•15•Updated May 27, 2025May 27, 2025
tlparse
Public
TORCH_LOGS parser for PT2
Rust
•
BSD 3-Clause "New" or "Revised" License
•16•37•30•2•Updated May 27, 2025May 27, 2025
kineto
Public
A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
HTML
•
Other
•184•818•56•11•Updated May 27, 2025May 27, 2025
opacus
Public
Training PyTorch models with differential privacy
neural-network differential-privacy privacy-preserving-machine-learning machine-learning deep-learning pytorch
Jupyter Notebook
•
Apache License 2.0
•369•1.8k•59•7•Updated May 27, 2025May 27, 2025
torchcodec
Public
PyTorch video decoding
Python
•
BSD 3-Clause "New" or "Revised" License
•36•564•57•17•Updated May 27, 2025May 27, 2025
vision
Public
Datasets, Transforms and Models specific to Computer Vision
machine-learning computer-vision
Python
•
BSD 3-Clause "New" or "Revised" License
•7.1k•17k•893•240•Updated May 27, 2025May 27, 2025
text
Public
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
•814•3.5k•290•56•Updated May 27, 2025May 27, 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
•694•2.7k•231•63•Updated May 27, 2025May 27, 2025
rfcs
Public
PyTorch RFCs (experimental)
Other
•76•131•1•39•Updated May 26, 2025May 26, 2025
tutorials
Public
PyTorch tutorials.
Jupyter Notebook
•
BSD 3-Clause "New" or "Revised" License
•4.2k•8.6k•149•15•Updated May 23, 2025May 23, 2025
pytorch_sphinx_theme
Public
PyTorch Sphinx Theme
CSS
•
MIT License
•140•99•22•8•Updated May 23, 2025May 23, 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
•372•2.8k•178•80•Updated May 23, 2025May 23, 2025
captum
Public
Model interpretability and understanding for PyTorch
interpretability feature-importance interpretable-ai interpretable-ml feature-attribution
Python
•
BSD 3-Clause "New" or "Revised" License
•519•5.2k•218•37•Updated May 22, 2025May 22, 2025
torchx
Public
TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iteration time for training/research and support for E2E production ML pipelines when you're ready.
components machine-learning airflow slurm pipelines ray aws-batch distributed-training python kubernetes
Python
•
Other
•131•366•71•48•Updated May 22, 2025May 22, 2025