diff --git a/docs/source/en/features/distributed_optimizers.md b/docs/source/en/features/distributed_optimizers.md index f95b23304..279bc8f9d 100644 --- a/docs/source/en/features/distributed_optimizers.md +++ b/docs/source/en/features/distributed_optimizers.md @@ -87,44 +87,42 @@ optim = DistGaloreAwamW( ## Plugin compatibility
Model/Feature | -Lamb | -GaLore | -Adafactor | -CAME | +Optimizer/Plugin | +Hybrid Parallel Plugin | +Low Level Zero Plugin | +Torch DDP Plugin | +Gemini Plugin | +Moe Hybrid Plugin | ||||||||||||||||||||||||||||||||||||||||||
---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|---|
Hybrid Parallel Plugin |
- ✔️ | +Lamb | ✔️ | ✔️ | ✔️ | +❌ | +❌ | |||||||||||||||||||||||||||||||||||||||||||||
Low Level Zero Plugin |
+ GaLore | ✔️ | -❌ | ✔️ | ✔️ | +❌ | +❌ | |||||||||||||||||||||||||||||||||||||||||||||
Torch DDP Plugin |
+ Adafactor | ✔️ | ✔️ | ✔️ | -✔️ | -|||||||||||||||||||||||||||||||||||||||||||||||
Gemini Plugin |
- ❌ | -❌ | ❌ | ❌ | ||||||||||||||||||||||||||||||||||||||||||||||||
Moe Hybrid Plugin |
- ❌ | -❌ | +CAME | +✔️ | +✔️ | +✔️ | ❌ | ❌ | ✔️ | ✔️ | ✔️ | -❌ | +✔️ | ❌ | ||||||||||||||||||||||||||||||||||||||
Model/Feature | -Lamb | -GaLore | -Adafactor | -CAME | +Optimizer/Plugin | +Hybrid Parallel Plugin | +Low Level Zero Plugin | +Torch DDP Plugin | +Gemini Plugin | +Moe Hybrid Plugin |
---|---|---|---|---|---|---|---|---|---|---|
Hybrid Parallel Plugin |
- ✔️ | +Lamb | ✔️ | ✔️ | ✔️ | +❌ | +❌ | |||
Low Level Zero Plugin |
+ GaLore | ✔️ | -❌ | ✔️ | ✔️ | +❌ | +❌ | |||
Torch DDP Plugin |
- ✔️ | +Adafactor | ✔️ | ✔️ | ✔️ | -|||||
Gemini Plugin |
- ❌ | -❌ | ❌ | ❌ | ||||||
Moe Hybrid Plugin |
- ❌ | -❌ | +CAME | +✔️ | +✔️ | +✔️ | ❌ | ❌ |