parallelism-strategies

Identify optimal TP, PP, DP, CP, and EP values for Megatron Bridge models.

2.8k|332|Updated Feb 25, 2026
One-click install
npx skills add https://github.com/NVIDIA/skills --skill parallelism-strategies
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: parallelism-strategies
Source: https://github.com/NVIDIA/skills/tree/main/skills/Megatron-Bridge/perf-techniques/parallelism-strategies
Command: npx skills add https://github.com/NVIDIA/skills --skill parallelism-strategies

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

Guides engineers in selecting and combining parallelism strategies (TP, PP, DP, SP, CP, EP) for Megatron Bridge to achieve scalable training performance and efficient resource usage.

Core Features & Use Cases

  • Provides decision guidance for dense and MoE models across single-node to multi-node deployments.
  • Offers sizing rules, hardware-topology mappings, and combined parallelism configurations to optimize throughput and memory.
  • Use case: when training a 236B MoE model on a 256-GPU cluster, apply recommended EP, PP, TP settings to maximize throughput while controlling interconnect overhead.

Quick Start

Configure and optimize TP, PP, DP, SP, and CP settings for Megatron Bridge across given hardware topology.

Frequently Asked Questions about parallelism-strategies

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I configure 3D parallelism for Megatron Bridge models?

Configuring 3D parallelism for Megatron Bridge involves identifying optimal starting values for TP, PP, DP, CP, and EP that satisfy specific sizing guidelines. It applies decision guidance to optimize throughput and memory across various hardware topologies.

What is the best way to choose parallelism strategies for a 236B MoE model on a multi-node cluster?

The best way to choose parallelism strategies for a 236B MoE model on a 256-GPU cluster is applying recommended EP, PP, and TP settings. This approach maximizes training throughput while effectively controlling InfiniBand interconnect overhead.

Does Megatron Bridge parallelism support both single-node NVLink and multi-node InfiniBand setups?

Yes, Megatron Bridge parallelism supports both single-node NVLink setups and multi-node InfiniBand clusters. It provides hardware-topology mappings and combined parallelism configurations tailored to optimize resource usage across these distinct environments.

How do I combine TP, PP, and DP settings to optimize distributed training throughput?

To optimize distributed training throughput, combine TP, PP, and DP settings using provided sizing rules and hardware-topology mappings. This combination balances computational load and memory efficiency to achieve scalable training performance.

When do I need expert parallelism (EP) and context parallelism (CP) in model parallelism configurations?

Expert parallelism (EP) and context parallelism (CP) are needed when configuring dense and MoE models for scalable training. They are specified alongside TP, PP, and DP to satisfy guidelines and control memory usage across large GPU clusters.