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
- [REQUEST] Support for XLA/TPU
- Fix: forbid repeated deepspeed.initialize on training objects
- AssertionError: no sync context manager is incompatible with gradientpartitioning logic of ZeRo stage 3
- Enabled configurable auto Tensor Parallelism (TP) for the inference of diverse models
- [BUG] Using and Building DeepSpeedCPUAdam
- [BUG] Fail to Resume From Checkpoint with Different GPU Number(Huggingface Trainer + Deepspeed)
- [BUG] Zero2 offload overflow
- [BUG] grad_norm and loss is nan when deepspeed==0.13.5 but ok with deepspeed==0.10.2
- Workflow for AutoTP
- [BUG] Deepspeed Zero 3 Inference InFlight Params with new HuggingFace Mixtral Model
- Docs
- Python not yet supported