LogoAwesome Skills
  • Search
  • Category
  • Tag
  • Blog
LogoAwesome Skills
LogoAwesome Skills

Discover Open-Source Agent Skills for AI Coding Assistants

Product

  • Search
  • Category
  • Tag
  • Blog

Resources

  • Claude Skill Docs
  • Antigravity Skills Docs

Tools

  • Claude Code
  • OpenCode
  • Cursor
  • Codex
  • Antigravity

Company

  • Privacy Policy
  • Terms of Service
  • Sitemap

©2026 Awesome Skills. All rights reserved.

Privacy PolicyTerms
Back to Skills

nemo-mbridge-perf-parallelism-strategies

Operational guide for choosing and combining parallelism strategies in Megatron Bridge, including sizing rules, hardware topology mapping, and combined parallelism configuration.

2,741stars319forksUpdated 7/31/2026
Data Science & ML#performance-tuning#gpu#machine-learning#nvidia#distributed-training

Security Assessment

Safe(96/100)
Security Score96/100

About nemo-mbridge-perf-parallelism-strategies

Parallelism Strategy Selection is an NVIDIA operational guide for choosing and combining parallelism strategies (tensor, data, pipeline, context, and expert parallelism) in Megatron Bridge. It solves the problem of deciding how to shard a model across GPUs: which degrees to use for a given model size, hardware topology, and sequence length, and how to trace an OOM or throughput regression back to a parallelism configuration change.

The skill provides decision tables by model size for both dense and MoE models (with concrete starting points for named models like DeepSeek-V2/V3, GLM-4.5, Qwen3, Kimi-K2), guidance by hardware topology (keep tensor parallelism within a single NVLink domain, use pipeline or data parallelism for cross-node scaling), and by sequence length (adding sequence and context parallelism as sequences grow). It gives copy-ready enablement snippets for 3D and 4D parallelism and MoE EP+PP configurations, and — notably — corrects a widespread sizing error: the minimum GPU count is PP times max(TP*CP, EP*ETP), not the full product PP*TP*CP*EP*ETP, because the attention and expert meshes overlap within a pipeline stage rather than multiply. It also covers implicit data-parallel sizing, scaling above the minimum, a memory-estimation worked example for a 70B model, and code anchors into the docs and model provider.

The target users are ML engineers and researchers configuring or debugging large-scale Megatron Bridge / NeMo training runs and sizing Slurm allocations. This is reference documentation and configuration guidance only — there are no scripts, destructive operations, credential handling, or network activity involved.

FAQ

What does this skill help me decide?

How to choose and combine tensor, data, pipeline, context, and expert parallelism degrees in Megatron Bridge based on model size, hardware topology, and sequence length, and how to trace OOM or regressions to a parallelism config change.

How does MoE parallelism differ from dense?

Because only a fraction of parameters are active per token, tensor parallelism is sized by active (not total) parameters and can often stay at 1 or 2, while expert parallelism becomes the primary scaling dimension and pipeline parallelism handles cross-node depth.

What is the correct minimum GPU count formula?

min_gpus = PP * max(TP*CP, EP*ETP). The common PP*TP*CP*EP*ETP full product is wrong because the attention and expert meshes share the same GPUs within a pipeline stage and overlap rather than multiply.

Where should tensor parallelism be kept?

Within a single NVLink domain. Use pipeline or data parallelism for cross-node scaling, since tensor parallelism across nodes is almost always a performance loss.

How do sequence length recommendations change?

Below 2K use standard TP+PP+DP; 2K-8K add sequence parallelism; 8K-32K add context parallelism of 2; and 32K+ use context parallelism of 4-8, considering a2a+p2p for large context parallel.

All Files

6 files
evals/evals.json1.5 KB
View
card.yaml2.9 KB
View
SKILL.md9.3 KB
View
BENCHMARK.md4.3 KB
View
skill.oms.sig4.7 KB
View
skill-card.md3.7 KB
View

Install nemo-mbridge-perf-parallelism-strategies

Download and extract the skill files to your .claude/skills/ directory.

Quick Setup:

  1. Copy the skill folder to .claude/skills/
  2. Claude will automatically detect and use the skill

Repository

nvidia/skills

Related Skills

auth

295

gsd-2-agent-framework

87

tailwind-patterns

48,337

tilegym-converting-cutile-to-julia

3,695