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 supported8 Subscribers
Add a CodeTriage badge to deepspeed
Help out
- Issues
- [REQUEST] MiCS vs Zero++ hpZ for Hybrid FSDP
- Flops Profiler
- Adding the new feature of FPDT
- [REQUEST] Replace reduce in ZERO 1/2/3 with reduce_scatter
- [BUG] Significant difference between using DeepSpeed and not using DeepSpeed
- [BUG] setting PYTORCH_CUDA_ALLOC_CONF in .deepspeed_env raise return code = -11
- [BUG] zero3 hang during inference, need to detach part of computational graph, .detach()/torch.no_grad do not work.
- configuration setting problems for parameters partitioning in training
- [REQUEST]Can we reload the model parameters after deepspeed.initialize?
- Unpin tests that previously used a pinned version of transformers
- Docs
- Python not yet supported