deepspeed
https://github.com/microsoft/deepspeed
Python
DeepSpeed is a deep learning optimization library that makes distributed training easy, efficient, and effective.
Triage Issues!
When you volunteer to triage issues, you'll receive an email each day with a link to an open issue that needs help in this project. You'll also receive instructions on how to triage issues.
Triage Docs!
Receive a documented method or class from your favorite GitHub repos in your inbox every day. If you're really pro, receive undocumented methods or classes and supercharge your commit history.
Python not yet supported17 Subscribers
Add a CodeTriage badge to deepspeed
Help out
- Issues
- Simple Example for Single-GPU Simple Model Training & Stand-Alone DeepSpeedCPUAdam
- How to use one's existing vertical MP with DeepSpeed
- A request for clarity around 3D Parallelism in DeepSpeed
- load_checkpoint nuances
- Model Parallel Example
- 3d parallel with megatorn-gpt2
- [website] general broken links
- some questions about zero-1 stage the source code
- Zero stage 2 : contiguous_gradients and "Reduce"
- torch.cuda.synchronize() takes a lot of time for the 10B model training with Zero 2
- Docs
- Python not yet supported