moe-optimization-workflow
OfficialStreamlined MoE training optimization workflow.
AuthorNVIDIA
Version1.0.0
Installs0
System Documentation
What problem does it solve?
Streamlines MoE training optimization by providing a repeatable workflow aligned with Megatron Bridge and Megatron Core principles to diagnose and fix bottlenecks.
Core Features & Use Cases
- Three Walls framework guidance for memory, communication, and compute bottlenecks.
- Parallel Folding and recompute strategy to decouple MoE and attention parallelism.
- Dispatcher selection and CUDA Graphs guidance to maximize throughput on scalable GPU setups.
Quick Start
Run the moe-optimization-workflow with Megatron-Bridge to guide end-to-end MoE training tuning.
Dependency Matrix
Required Modules
None requiredComponents
Standard package💻 Claude Code Installation
Recommended: Let Claude install automatically. Simply copy and paste the text below to Claude Code.
Please help me install this Skill: Name: moe-optimization-workflow Download link: https://github.com/NVIDIA/skills/archive/main.zip#moe-optimization-workflow Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
Agent Skills Search Helper
Install a tiny helper to your Agent, search and equip skill from 471,000+ vetted skills library on demand.