What problem does it solve?
This Skill provides a production-grade training framework to efficiently train multi-billion-parameter language models by orchestrating tensor, pipeline, sequence, context and expert parallelism to maximize GPU utilization, reduce memory pressure, and stabilize large-scale pretraining runs.
Core Features & Use Cases
- Scalable Parallelism: Guides configuration of tensor, pipeline, context, data and expert parallelism for models ranging from 2B to 462B parameters.
- Performance Optimizations: Covers FP8/FP16/BF16 precision, Flash Attention, Transformer Engine integration, sequence parallelism and micro-batch tuning to improve MFU on H100/A100 clusters.
- MoE & Production Workflows: Includes Mixture-of-Experts recipes, checkpointing strategies, monitoring guidance and multi-node production deployment examples for LLaMA, Mixtral, Nemotron and other large models.
- Use Case: Prepare and launch a production LLaMA-3 70B pretraining job across 64 H100 GPUs with TP+PP+CP for high throughput and stable convergence.
Quick Start
Launch a distributed LLaMA 70B pretraining run on 64 H100 GPUs with Megatron-Core using FP8 and Transformer Engine optimizations.