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 supported21 Subscribers
Add a CodeTriage badge to deepspeed
Help out
- Issues
- [Apple Silicon Support Phase 1] Add Metal FusedAdam kernel and CPU Adam build for Apple Silicon
- feat(opsd): share prompt prefill across rollout samples
- [BUG] ZeRO-1/2 fail mapping zero-sized parameter inside a partition
- Fix AutoTP metadata updates for unsharded modules
- perf(rollout): add HybridEngine rollout profiling
- Skip zero-sized parameters in HP fragment mapping
- [Ulysses SP] process-wide _ulysses_num_kv_heads global breaks a second model with a different head count in the same process
- HF transformers main injects 'embedding_rowwise' into tp_plan for tied-embedding models; AutoTP rejects the whole plan
- [AutoTP] update_mp_params shards vision-tower attributes (patch_embed.embed_dim) without sharding weights, breaking multimodal forward
- Register native pinned host memory with CUDA for GPU DMA
- Docs
- Python not yet supported