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 supported21 Subscribers
Add a CodeTriage badge to deepspeed
Help out
- Issues
- docs: add Simplified Chinese translation (README.zh.md)
- Fix ZeRO-3 checkpoint tag selection during weight reconstruction
- Recognize a norm by its shape, not by whether its class name was listed
- [BUG] Memory leak in loss backward when enabled torch activation checkpoint & zero3 & single-model-multi-branch network
- tests: allow testing an installed DeepSpeed package
- [Bug] AutoEP FP32 gradient clipping averages norms across ranks with different expert ownership
- Can't install in Python 3.14.6 with GCC
- Count each expert once in the fp32 gradient-clipping norm
- Refresh the whole captured input before a CUDA graph replay
- Fix AutoEP global L2 clipping for sharded expert ownership
- Docs
- Python not yet supported