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 supported15 Subscribers
Add a CodeTriage badge to deepspeed
Help out
- Issues
- Introduce all_reduce_hook to support gradient aggregation across replica groups.
- [BUG] Bug while using deepspeed with TRL with vLLM
- [REQUEST] Expose TP/DP/PP Rank as Environment Variable for integration with low Level Profiling Tools
- [REQUEST] Add support for non-__dict__ outputs such as MinkowskiEngine SparseTensor in ZeRO Stage 3 (DeepSpeed v0.9.2)
- Add EXAONE 4.0 model support for DeepSpeed inference v2 @
- Add world-size getter in Engine
- Configures workflow for offline unit tests
- [BUG]Installation Error on Torch while Torch is Installed
- [REQUEST] Switch to a unified process group management framework for DP/TP/PP/EP/SP
- [BUG] DeepCompile: MemoryProfiling error /pytorch/build/aten/src/ATen/RegisterCUDA.cpp:7280: SymIntArrayRef expected to contain only concrete integers
- Docs
- Python not yet supported