training-llms-megatron

Optimize Megatron-Core LLM training workflows across multi-GPU clusters.

Updated Mar 30, 2026
One-click install
npx skills add https://github.com/KappTech88/AI-RESEARCH-SKILLS-MCP --skill training-llms-megatron-kapptech88
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: training-llms-megatron
Source: https://github.com/KappTech88/AI-RESEARCH-SKILLS-MCP/tree/main/skills/megatron-core
Command: npx skills add https://github.com/KappTech88/AI-RESEARCH-SKILLS-MCP --skill training-llms-megatron-kapptech88

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes references (resource) components.

What problem does it solve?

Efficiently train very large language models using Megatron-Core with advanced parallelism, enabling scalable infrastructure for cutting-edge models.

Core Features & Use Cases

  • Advanced parallelism: tensor, pipeline, sequence, context, and expert parallelism to train multi-billion-parameter models.
  • Production-ready workflows: guidance for Nemotron, LLaMA, and DeepSeek deployments with MFU optimization on H100.
  • Reference configurations: end-to-end training recipes for GPT, LLaMA, and Mixtral architectures.

Quick Start

Launch Megatron-Core training with your chosen model size and hardware setup.

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 large language models with Megatron-Core using tensor and pipeline parallelism?

To train large language models with Megatron-Core, you apply tensor, pipeline, sequence, context, and expert parallelism across multi-GPU clusters. This Skill provides reference configurations to optimize end-to-end workflows for models ranging from 2B to 462B parameters.

What parallelism strategies are needed to optimize MFU on H100 clusters for LLM training?

Optimizing MFU on H100 clusters for LLM training requires combining tensor, pipeline, sequence, context, and expert parallelism. The Skill provides production-ready workflows and reference recipes to maximize hardware utilization for GPT, LLaMA, and Mixtral architectures.

Can I use this Megatron-Core training workflow for DeepSeek and Nemotron deployments?

Yes, you can use this Megatron-Core training workflow for DeepSeek and Nemotron deployments. The Skill includes production-ready guidance and end-to-end training recipes specifically tailored for these architectures alongside LLaMA configurations.

Do I need transformer-engine and apex to run multi-billion-parameter model training with Megatron-Core?

Yes, you need transformer-engine and apex along with megatron-core and torch to run multi-billion-parameter model training. These dependencies are required to execute the advanced parallelism configurations and achieve optimal MFU on H100 hardware.

What is the best way to configure expert parallelism for MoE models in distributed training?

The best way to configure expert parallelism for MoE models is by using the provided Mixtral reference recipes. These configurations leverage Megatron-Core to distribute expert layers across multi-GPU clusters, maximizing training efficiency for large-scale architectures.

Why does distributed training fail to scale efficiently beyond standard data parallelism for 400B parameter models?

Standard data parallelism fails to scale efficiently for 400B parameter models due to severe memory bottlenecks. Combining data, tensor, pipeline, context, and expert parallelism using Megatron-Core resolves these constraints, enabling successful 462B parameter training.