A CPU+GPU Profiling library that provides access to timeline traces and hardware performance counters.
OTHER License
Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.
Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of c...
Run PyTorch LLMs locally on servers, desktop and mobile
torchao: PyTorch Architecture Optimization (AO). Performant kernels that work with PyTorch.
PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT
Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications.
Lightwood is Legos for Machine Learning.
State-of-the-Art Deep Learning scripts organized by models - easy to train and deploy with reprod...
High-level library to help with training and evaluating neural networks in PyTorch flexibly and t...
TorchX is a universal job launcher for PyTorch applications. TorchX is designed to have fast iter...
A modular, primitive-first, python-first PyTorch library for Reinforcement Learning.
Train, Evaluate, Optimize, Deploy Computer Vision Models via OpenVINO™
Enabling PyTorch on XLA Devices (e.g. Google TPU)