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
- Regarding train_batch_size
- [RuntimeError: Connection reset by peer] When scaling up training jobs
- RuntimeError: Could not find llvm-config. Please install llvm-{8, 9, 10}-dev
- stuck in trying to combine PP with DeepSpeed
- Inference with an autoregressive Pipeline parallel model
- new_grad_tensor.copy_(param.grad.view(-1)) AttributeError: 'NoneType' object has no attribute 'view'
- Model doesn't learn with fp16.enabled = true: [Rank 0] Overflow detected. Skipping step. Attempted loss scale: 1, reducing to 1
- Activation Checkpointing breaks for some layers in PipelineModule
- Simple Example for Single-GPU Simple Model Training & Stand-Alone DeepSpeedCPUAdam
- How to use one's existing vertical MP with DeepSpeed
- Docs
- Python not yet supported