What it does
Composer is a toolkit for training deep-learning models, from language models to image generators. It's built on PyTorch, a popular AI programming framework, and handles hard parts such as spreading training across many GPUs and computers.
It includes automatic resuming after a failed run, automatic memory tuning to avoid crashes, and a set of speed-up techniques. Its makers say these can make some training jobs several times faster.
It connects to experiment-tracking tools and cloud storage.
Who it's for
Machine-learning engineers and researchers training models at scale.
Can I use it without coding?
Developers only. It's a Python library for people comfortable with PyTorch. New to this? Start here.
Can I run it?
- Cost
- FreeNothing to pay or sign up for.
- Computer
- Works best on a powerful computer or one with a graphics card (GPU)
- Privacy
- Runs on your own device: your data stays with you
- API key
- Not needed
Try it
Using Claude Code, Cursor or another AI coding assistant? Paste the prompt and it will walk you through the setup.New to this?
Before you install
- Only download from the official GitHub page. Copies with similar names exist for many popular projects.
- License (Apache-2.0): Permissive: free to use, change and share, including for business.
- More on staying safe
Where people are talking about it
- Composer: A new PyTorch library to train models 4x faster with better algorithmsHacker News · Mar 2022