Top pick
Microsoft library for extreme-scale model training: ZeRO optimizer states, pipeline parallelism, and inference kernels.
trainingdistributedlarge-modelspytorch
Browse & filter
Filter by platform, license text, maturity, maintenance cadence, and editorial tags like privacy-focused or self-hosted. Search matches names, summaries, tags, and use cases.
6 tools match your filters
Top pick
Hugging Face library for diffusion models: training, inference, schedulers, and community pipelines in PyTorch.
diffusiongenerativepytorchimages
Also strong
Hugging Face library to run PyTorch training on CPU, single GPU, multi-GPU, or TPU with minimal code changes.
distributedtrainingpytorchhuggingface
Also strong
Physics-ML / scientific deep learning framework: neural operators, PINNs, and domain-parallel training on GPUs.
physics-mlsimulationpytorchgputaaft-repositories
State-of-the-art pretrained models for PyTorch, TensorFlow, and JAX.
nlppytorchmodels
Honorable mention
High-level deep learning library on PyTorch.
deep-learningpytorchcourse