perf-parallelism-strategies

Select and configure parallelism strategies for large-scale model training in Megatron Bridge.

852|445|Updated May 21, 2025
One-click install
npx skills add https://github.com/NVIDIA-NeMo/Megatron-Bridge --skill perf-parallelism-strategies
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: perf-parallelism-strategies
Source: https://github.com/NVIDIA-NeMo/Megatron-Bridge/tree/main/skills/perf-parallelism-strategies
Command: npx skills add https://github.com/NVIDIA-NeMo/Megatron-Bridge --skill perf-parallelism-strategies

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

This Skill helps users understand how to effectively choose and combine various parallelism strategies in Megatron Bridge to improve training efficiency and scalability.

Core Features & Use Cases

  • Parallelism Strategy Selection: Guides users in selecting appropriate data, tensor, pipeline, expert, and sequence parallelism based on model size, hardware topology, and sequence length.
  • Profiling and Sizing Advice: Provides heuristic guidelines and minimum GPU counts for different parallelism configurations, helping users plan resources efficiently.
  • Technical Resource: Explains the technical constraints, performance considerations, and implementation details necessary for advanced parallelism setup.

Quick Start

Configure the tensor_model_parallel_size, pipeline_model_parallel_size, and sequence_parallel settings in the model provider based on your hardware topology and model size to optimize training performance.

Frequently Asked Questions about perf-parallelism-strategies

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

FAQPage Schema
How do I choose the right parallelism strategy for large-scale model training?

To choose a parallelism strategy, evaluate your model size, sequence length, and hardware topology. Guide the selection by combining data, tensor, pipeline, expert, and sequence parallelism to ensure optimal resource utilization and performance scalability.

What is the difference between tensor parallel and pipeline parallel configurations?

Tensor parallel partitions individual model layers across GPUs, while pipeline parallel splits sequential model layers across devices. Aligning these parallelism methods with system constraints ensures efficient resource utilization and performance scalability during large-scale training.

How do I configure tensor and pipeline parallel size in Megatron Bridge?

Configure parallelism by setting tensor_model_parallel_size, pipeline_model_parallel_size, and sequence_parallel in the model provider. Base these settings on your specific hardware topology and model size to optimize training performance.

What is the minimum GPU count required for different parallelism configurations?

Minimum GPU counts for parallelism configurations depend on heuristic guidelines based on your model size and hardware topology. Profiling and sizing advice helps plan resources efficiently by providing specific hardware constraints for different setups.

When should I use sequence parallelism in distributed training?

Use sequence parallelism in distributed training when working with long sequence lengths. Aligning this method with your hardware topology and model size constraints ensures optimal resource utilization and performance scalability.

Does perf-parallelism-strategies support expert parallelism for large models?

Yes, the strategy selection guides configuring expert parallelism alongside data, tensor, pipeline, and sequence parallelism. This ensures optimal resource utilization and performance scalability by aligning parallelism methods with system constraints.