training-llms-megatron

Train large language models with distributed parallelism across multi-GPU clusters.

Updated Feb 15, 2026
One-click install
npx skills add https://github.com/ricable/mcai --skill training-llms-megatron-ricable
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: training-llms-megatron
Source: https://github.com/ricable/mcai/tree/main/.agents/skills/training-llms-megatron
Command: npx skills add https://github.com/ricable/mcai --skill training-llms-megatron-ricable

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill includes references (resource) components.

What problem does it solve?

This Skill removes the complexity of training very large language models by turning distributed pretraining, memory planning, and performance tuning into a repeatable workflow.

Core Features & Use Cases

  • Large-Scale Model Training: Plan and run dense or mixture-of-experts models from billions to hundreds of billions of parameters.
  • Parallelism Strategy Guidance: Choose the right mix of tensor, pipeline, sequence, context, and expert parallelism for your cluster.
  • Performance Optimization: Improve throughput, memory usage, and model flop utilization with practical H100 and A100 recommendations.
  • Production Recipes: Apply tested configurations for GPT, LLaMA, Mixtral, DeepSeek, and other real-world training scenarios.
  • Use Case: A research team preparing a 70B or 405B pretraining run can use this Skill to select a GPU layout, tune batch sizes, and avoid memory or communication bottlenecks.

Quick Start

Use this skill to design and validate a Megatron-Core training configuration for your target model size, hardware cluster, and parallelism strategy.

Frequently Asked Questions about training-llms-megatron

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

FAQPage Schema
How do I configure distributed training for large language models using Megatron-Core?

Distributed training for large language models using Megatron-Core is configured by selecting a mix of tensor, pipeline, sequence, context, and expert parallelism. This skill provides benchmark-backed recipes and production configuration examples to validate your cluster setup.

What is the best way to plan GPU memory and parallelism for a 70B pretraining run?

The best way to plan GPU memory and parallelism for a 70B pretraining run is to apply tested configurations for GPT and LLaMA architectures. This skill offers memory optimization guidance and throughput tuning to avoid communication bottlenecks across multi-GPU and multi-node workflows.

Does Megatron-Core support mixture of experts training for models like Mixtral and DeepSeek?

Yes, Megatron-Core supports mixture of experts training for models like Mixtral and DeepSeek. It provides specific expert parallel requirements and production recipes to manage high-throughput pretraining for these specialized architectures.

Can I use this approach to optimize model flop utilization on H100 and A100 clusters?

You can optimize model flop utilization on H100 and A100 clusters by applying practical performance tuning recommendations. This skill helps improve throughput and memory usage during dense or mixture-of-experts model training at scale.

When do I need to use sequence and context parallelism alongside tensor and pipeline parallelism?

Sequence and context parallelism are needed alongside tensor and pipeline parallelism when training very large language models that exceed single-device memory limits. This skill helps you choose the right parallelism strategy combination for your specific hardware cluster.

Why does my large-scale LLM training run face communication bottlenecks across multiple nodes?

Large-scale LLM training runs face communication bottlenecks across multiple nodes due to suboptimal parallelism strategies and batch sizes. This skill provides benchmark-backed recipes to tune configurations and eliminate memory or communication performance issues.