nemo-mbridge-perf-tp-dp-comm-overlap

Community

Unlock TP/DP/PP comm overlap for faster training

Authorsayalinvidia
Version1.0.0
Installs0

System Documentation

What problem does it solve?

Enables and validates Tensor-Parallel, Data-Parallel, and Pipeline-Parallel communication overlap in Megatron-Bridge to boost training throughput and efficiency.

Core Features & Use Cases

  • Provides configuration knobs for TP, DP, and PP overlap in Megatron-Bridge training.
  • Includes code anchors and launch-time environment tuning guidance.
  • Offers verification steps and pitfalls to ensure safe, correct overlap behavior.
  • Use Case: Optimizing throughput for large-scale transformer training and diagnosing related performance regressions.

Quick Start

Enable TP/DP/PP communication overlap by applying the recommended CommOverlapConfig settings in Megatron-Bridge and verify the wiring with the provided unit tests.

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