nemo-mbridge-perf-expert-parallel-overlap

Community

Speed up Megatron-Bridge MoE with EP overlap.

Authorsayalinvidia
Version1.0.0
Installs0

System Documentation

What problem does it solve?

Overlap expert-parallel token dispatch/collect across multiple GPUs to hide communication costs and improve throughput in Megatron-Bridge MoE workloads.

Core Features & Use Cases

  • Overlaps expert-parallel dispatch/collect with FFN compute to reduce per-step latency.
  • Supports both alltoall and flex dispatchers; applicable to large MoE models requiring throughput tuning.
  • Provides guidance for correctness-first validation and performance evaluation in Megatron-Bridge deployments.

Quick Start

Run the Megatron-Bridge performance setup to enable EP overlap testing on a representative MoE model.

Dependency Matrix

Required Modules

None required

Components

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: nemo-mbridge-perf-expert-parallel-overlap
Download link: https://github.com/sayalinvidia/sayali-skills-test/archive/main.zip#nemo-mbridge-perf-expert-parallel-overlap

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 510,000+ vetted skills library on demand.