pytorch

All

80 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
•23k•85k•14k•1.1k•Updated Dec 31, 2024Dec 31, 2024
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
•503•5k•212•40•Updated Dec 31, 2024Dec 31, 2024
executorch
Public
On-device AI across mobile, embedded and edge for PyTorch
machine-learning mobile embedded deep-learning neural-network gpu tensor
C++
•
Other
•409•2.3k•270•662•Updated Dec 31, 2024Dec 31, 2024
FBGEMM
Public
FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
C++
•
Other
•520•1.2k•39•384•Updated Dec 31, 2024Dec 31, 2024
torchcodec
Public
PyTorch video decoding
Python
•
BSD 3-Clause "New" or "Revised" License
•14•175•38•9•Updated Dec 31, 2024Dec 31, 2024
torchtune
Public
PyTorch native post-training library
Python
•
BSD 3-Clause "New" or "Revised" License
•469•4.5k•238•53•Updated Dec 31, 2024Dec 31, 2024
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
•355•2.6k•190•15•Updated Dec 31, 2024Dec 31, 2024
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
•455•2k•61•272•Updated Dec 31, 2024Dec 31, 2024
vision
Public
Datasets, Transforms and Models specific to Computer Vision
machine-learning computer-vision
Python
•
BSD 3-Clause "New" or "Revised" License
•7k•16k•857•223•Updated Dec 31, 2024Dec 31, 2024
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
•812•3.5k•281•53•Updated Dec 31, 2024Dec 31, 2024
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
•155•1.2k•207•30•Updated Dec 31, 2024Dec 31, 2024
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
•664•2.6k•214•52•Updated Dec 31, 2024Dec 31, 2024
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
•84•83•219•169•Updated Dec 31, 2024Dec 31, 2024
cppdocs
Public
PyTorch C++ API Documentation
HTML
•34•211•7•3•Updated Dec 31, 2024Dec 31, 2024
tnt
Public
A lightweight library for PyTorch training tools and utilities
machine-learning reinforcement-learning deep-learning neural-network python pytorch
Python
•
Other
•278•1.7k•5•78•Updated Dec 31, 2024Dec 31, 2024
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
•1•7•2•0•Updated Dec 31, 2024Dec 31, 2024
functorch
Public
functorch is JAX-like composable function transforms for PyTorch.
gradients hessians pytorch
Jupyter Notebook
•
BSD 3-Clause "New" or "Revised" License
•102•1.4k•232•18•Updated Dec 31, 2024Dec 31, 2024
tutorials
Public
PyTorch tutorials.
Jupyter Notebook
•
BSD 3-Clause "New" or "Revised" License
•4.1k•8.3k•157•30•Updated Dec 30, 2024Dec 30, 2024
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
•188•1.7k•193•89•Updated Dec 30, 2024Dec 30, 2024
botorch
Public
Bayesian optimization in PyTorch
Jupyter Notebook
•
MIT License
•410•3.1k•57•19•Updated Dec 30, 2024Dec 30, 2024
torchtitan
Public
A PyTorch native library for large model training
Python
•
BSD 3-Clause "New" or "Revised" License
•229•2.8k•55•35•Updated Dec 27, 2024Dec 27, 2024
torchchat
Public
Run PyTorch LLMs locally on servers, desktop and mobile
local pytorch llm
Python
•
BSD 3-Clause "New" or "Revised" License
•229•3.4k•102•24•Updated Dec 27, 2024Dec 27, 2024
xla
Public
Enabling PyTorch on XLA Devices (e.g. Google TPU)
compiler xla deep-learning pytorch
C++
•
Other
•487•2.5k•593•148•Updated Dec 27, 2024Dec 27, 2024
builder
Public
Continuous builder and binary build scripts for pytorch
Python
•
BSD 2-Clause "Simplified" License
•227•343•79•70•Updated Dec 24, 2024Dec 24, 2024
pytorch.github.io
Public
The website for PyTorch
HTML
•
BSD 3-Clause "New" or "Revised" License
•295•233•156•60•Updated Dec 24, 2024Dec 24, 2024
kineto
Public
A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
HTML
•
Other
•170•754•42•5•Updated Dec 23, 2024Dec 23, 2024
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
•293•887•67•51•Updated Dec 23, 2024Dec 23, 2024
serve
Public
Serve, optimize and scale PyTorch models in production
docker kubernetes cpu metrics gpu optimization serving mlops machine-learning deep-learning
Java
•
Apache License 2.0
•866•4.3k•378•47•Updated Dec 20, 2024Dec 20, 2024
tensordict
Public
TensorDict is a pytorch dedicated tensor container.
Python
•
MIT License
•77•859•34•23•Updated Dec 20, 2024Dec 20, 2024
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
•348•1.7k•61•12•Updated Dec 20, 2024Dec 20, 2024