Train larger models with fewer active experts.
Guides the implementation and training of sparse mixture-of-experts models with routing, load balancing, and expert parallelism.
When to use it
Use when training larger models with limited compute, scaling model capacity, or specializing experts for different domains, tasks, or languages.
Give it a model-training task in a project; it provides mixture-of-experts implementation and training configuration.
What you provide
The model implementation imports torch and its neural-network module.
This skill
deepspeed
Installs DeepSpeed for MoE training