moe-long-context
OfficialGuidance for scaling MoE long-context training.
AuthorNVIDIA
Version1.0.0
Installs0
System Documentation
What problem does it solve?
Long-context MoE training presents memory and throughput challenges as sequence lengths grow; this guide helps engineers optimize CP sizing, dispatcher layouts, and recompute strategies to sustain performance.
Core Features & Use Cases
- Provides CP scaling rules for MoE training on DSV3, Qwen3, and Qwen3-Next on NV hardware.
- Recommends selective recompute, CUDA graphs advice, and offload strategies to manage memory and compute.
- Includes representative config patterns and practical tips for success in long-context MoE workloads.
Quick Start
Configure your Megatron Bridge MoE training with the recommended CP, dispatcher, and selective recompute settings to evaluate throughput.
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-long-context Download link: https://github.com/NVIDIA/skills/archive/main.zip#moe-long-context 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.