torchmetrics

py-motmetrics

Benchmark multiple object trackers (MOT) in Python

pytorch_geometric_signed_directed

PyTorch Geometric Signed Directed is a signed/directed graph neural network extension library for...

NeuroTorch

Biologically based dynamics for machine learning and deep learning tools for neuroscience.

pytorch-lightning

Pretrain, finetune and deploy AI models on multiple GPUs, TPUs with zero code changes.

pytorch-dist

tensorwatch

Debugging, monitoring and visualization for Python Machine Learning and Data Science

CCSD

Combinatorial Complex Score-based Diffusion model using stochastic differential equations

accelerate

🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed conf...

refinery-python-sdk

Official Python SDK for Kern AI refinery.

ignite

High-level library to help with training and evaluating neural networks in PyTorch flexibly and t...

litgpt

20+ high-performance LLM implementations with recipes to pretrain, finetune and deploy at scale.

lightning-thunder

Make PyTorch models up to 40% faster! Thunder is a source to source compiler for PyTorch. It enab...

diffusers-torchao

End-to-end recipes for optimizing diffusion models with torchao and diffusers (inference and FP8 ...