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 supported9 Subscribers
Add a CodeTriage badge to deepspeed
Help out
- Issues
- Activation Checkpointing conflicts with Weight Sharing
- [REQUEST] When training a FP16 model, the ability to set some of the layers to FP32
- [Bug] - summing over DP instead of MP
- Question about fp16 dynamic loss scale overflow in DeepSpeed
- [BUG]DeepSpeed Comm. Backend not compatible with outside torch.distributed module
- [BUG] Recommended way to implement EMA
- Reshape ZeroStage=0 FP16 Checkpoint
- DeepSpeed initialization with GNN-like model
- Is it possible to not print the deepspeed configuration? It is too long.
- [BUG] zero_to_fp32 ordering files incorrectly for combining shards
- Docs
- Python not yet supported