perf-moe-optimization-workflow

Official

Optimize MoE training for maximum throughput.

AuthorNVIDIA-NeMo
Version1.0.0
Installs0

System Documentation

What problem does it solve?

This Skill provides a systematic workflow for optimizing the training performance of Mixture-of-Experts models, helping users enhance throughput and reduce bottlenecks.

Core Features & Use Cases

  • Performance Tuning: Guides users through memory, communication, and compute optimizations for MoE models.
  • Workflow Phases: Offers a clear, phased approach from fitting models into memory to profiling and scaling.
  • Use Case: Ideal for deep learning engineers tuning large-scale MoE training jobs on GPU clusters, such as Megatron-LM, to improve efficiency and scalability.

Quick Start

Describe the workflow steps for diagnosing and enhancing MoE model training performance.

Dependency Matrix

Required Modules

None required

Components

references

💻 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: perf-moe-optimization-workflow
Download link: https://github.com/NVIDIA-NeMo/Megatron-Bridge/archive/main.zip#perf-moe-optimization-workflow

Please download this .zip file, extract it, and install it in the .claude/skills/ directory.
View Source Repository

Agent Skills Search Helper

Install a tiny helper to your Agent, search and equip skill from 471,000+ vetted skills library on demand.