training-llms-megatron

Train multi-billion parameter language models with distributed parallelism on GPU clusters.

2|Updated Apr 12, 2026
One-click install
npx skills add https://github.com/Clay-HHK/claude-config --skill training-llms-megatron-clay-hhk
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: training-llms-megatron
Source: https://github.com/Clay-HHK/claude-config/tree/main/skills/AI-research-SKILLs/08-distributed-training/megatron-core
Command: npx skills add https://github.com/Clay-HHK/claude-config --skill training-llms-megatron-clay-hhk

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes references (resource) components.

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.

Frequently Asked Questions about training-llms-megatron

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

FAQPage Schema
How do I train a multi-billion parameter LLM using distributed parallelism on GPU clusters?

To train multi-billion parameter LLMs, you orchestrate tensor, pipeline, sequence, context, and expert parallelism across multi-node GPU clusters. This maximizes GPU utilization and reduces memory pressure for models ranging from 2B to 462B parameters.

What's the best way to configure Mixture-of-Experts for large-scale pretraining?

Configuring Mixture-of-Experts for pretraining involves applying expert parallelism alongside tensor and pipeline parallelism. This distributes model components efficiently to maximize throughput and stabilize large-scale training runs.

Can I use FP8 precision and Flash Attention to improve MFU on H100 clusters?

Yes, you can use FP8 precision and Flash Attention to improve Model Flops Utilization (MFU). Integrating Transformer Engine optimizations on H100 clusters significantly enhances throughput during distributed training.

How do I launch a LLaMA 70B pretraining job across 64 H100 GPUs?

To launch a LLaMA 70B pretraining job across 64 H100 GPUs, apply tensor, pipeline, and context parallelism (TP+PP+CP) combined with FP8 optimizations to achieve high throughput and stable convergence.

When should I use tensor and pipeline parallelism during model training?

Use tensor and pipeline parallelism when training models exceeding single-GPU memory limits. Orchestrating these parallelism techniques reduces memory pressure and maximizes GPU utilization for multi-billion parameter pretraining.