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 supported18 Subscribers
Add a CodeTriage badge to deepspeed
Help out
- Issues
- Deepspeed inference support GPT BigCode (bigcode/starcoder, bigcode/gpt_bigcode-santacoder, etc.)
- where is pretrain_zeropp_gpt.py
- [BUG]Zero stage3 can not save model weights correctly!
- [BUG]- Training on multiple GPU's gives error in accelerator.backward(loss)
- [BUG] Partitioned activation checkpointing does not reduce memory usage
- Zero cpu offloading is not working
- [BUG] KeyError in stage_1_and_2.py when training dreambooth with deepspeed (in kohya_ss)
- [BUG] Tensor are not on the same device when enable cpu activation offload
- [REQUEST] DeepSpeed Zero3 swap off gradients unnecessarily when swap_optimizer is True
- Add FALCON-40B Inference-Kernel Support
- Docs
- Python not yet supported