functorch is JAX-like composable function transforms for PyTorch.
BSD-3-CLAUSE License
Run PyTorch LLMs locally on servers, desktop and mobile
A JavaScript library like PyTorch, with GPU acceleration.
A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating poin...
A flexible package for multimodal-deep-learning to combine tabular data with text and images usin...
Constrained optimization toolkit for PyTorch
Torchhd is a Python library for Hyperdimensional Computing and Vector Symbolic Architectures
[ECCV 2022] SimpleRecon: 3D Reconstruction Without 3D Convolutions
Pytorch framework for doing deep learning on point clouds.
Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications.
Rust bindings for the C++ api of PyTorch.
torchao: PyTorch Architecture Optimization (AO). Performant kernels that work with PyTorch.
Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of c...
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
The Incredible PyTorch: a curated list of tutorials, papers, projects, communities and more relat...
Easy Hypernetworks in Pytorch and Jax