mlm-bridge-training

Coordinate Megatron-LM MLM pretraining and Megatron Bridge training experiments on multi-GPU setups.

2.8k|332|Updated Feb 25, 2026
One-click install
npx skills add https://github.com/NVIDIA/skills --skill mlm-bridge-training
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: mlm-bridge-training
Source: https://github.com/NVIDIA/skills/tree/main/skills/Megatron-Bridge/mlm-bridge-training
Command: npx skills add https://github.com/NVIDIA/skills --skill mlm-bridge-training

SYSTEM DOCUMENTATION & REQUIREMENTS

What problem does it solve?

Enables running and comparing Megatron-LM MLM pretraining and Megatron Bridge training with mock or real data, supporting reproducible experiments across multi-GPU setups.

Core Features & Use Cases

  • Correlation testing between MLM and Bridge training runs to verify consistency.
  • Recipe-driven experimentation with multiple configurations and translation support for Bridge.
  • Multi-GPU / distributed training guidance for scalable benchmarks and real-world workloads.

Quick Start

Run the provided recipes to execute MLM pretraining and Bridge training side-by-side on your hardware.

Frequently Asked Questions about mlm-bridge-training

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

FAQPage Schema
How do I compare Megatron-LM MLM and Bridge training losses?

You can compare Megatron-LM MLM and Bridge training losses by running provided recipes side-by-side on multi-GPU setups. This executes both workflows to generate reproducible experiment outputs for direct loss comparison and correlation testing.

Can I use mock data for Megatron-LM distributed training experiments?

Yes, you can use mock data for Megatron-LM distributed training experiments. The recipes support mock data inputs alongside real data, allowing you to verify configuration translations and test training consistency without preparing a full dataset.

What is Bridge training configuration translation in Megatron-LM?

Bridge training configuration translation in Megatron-LM is the documented process of mapping parameters between MLM pretraining and Bridge training recipes. It ensures consistent, reproducible training workflows across different experimental configurations.

Does correlation testing verify consistency between MLM and Bridge training runs?

Yes, correlation testing verifies consistency between MLM and Bridge training runs by comparing their losses and configuration translations. This validates that training results remain reproducible across different multi-GPU environments and recipe setups.

How do I set up multi-GPU environments for Megatron-LM Bridge training?

To set up multi-GPU environments for Megatron-LM Bridge training, follow the explicit environment setup steps provided in the recipes. These include configuring CLI arguments and distributed training settings to execute scalable benchmarks.