#76 · Primary category: Deep Learning Frameworks
composer
Supercharge Your Model Training
Project last updated:04/29/26
GitHub Stars
5.5K
Forks
467
Contributors
140
License
Apache-2.0
Why we included this project
Composer is the kind of training library teams reach for when their PyTorch workloads outgrow a single GPU but they don't want to hand-build distributed training plumbing. It wraps the training loop in a Trainer abstraction that handles parallelism, sharded checkpoints, and memory optimization, so you can focus on your model and experiments rather than low-level cluster mechanics. The callback and event system is genuinely flexible: researchers can inject custom logic at any point in the loop, and the built-in speedup algorithms apply research-backed efficiency methods that have meaningfully cut training time for common models such as ResNet and BERT. Teams training LLMs, diffusion models, transformers, or CNNs on multi-node setups will find it a practical foundation, and elastic checkpointing plus auto-resumption makes long or interrupted runs far less painful. It is a solid pick if you are weighing a maintained training framework versus assembling your own stack from raw PyTorch primitives.
Articles for this project
No articles for this project yet.
To suggest a topic or contribute an article, contact us.
Related projects in this category
tensorflow
An Open Source Machine Learning Framework for Everyone
pytorch
Tensors and Dynamic neural networks in Python with strong GPU acceleration
keras
Deep Learning for humans
nanoGPT
The simplest, fastest repository for training/finetuning medium-sized GPTs.
ray
Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.