NVIDIA-NeMo/Megatron-Bridge

Training library for Megatron-based models with bidirectional Hugging Face conversion capability

900Stars on the repository
38Mods indexed here, across every type
todayLast push, which is what freshness is scored on
Apache-2.0Licence, which decides whether bodies are shown

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

Guide for adding support for new LLM or VLM models in Megatron-Bridge. Covers bridge, provider, recipe, tests, docs, and examples.

not rated 893 +3 2d ago A 36 tokens original Apache-2.0

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

Dev environment setup for Megatron Bridge — container-based development, uv package management, lockfile regeneration, adding dependencies, Slurm container usage, and common build pitfalls.

not rated 893 +3 2d ago A 38 tokens original Apache-2.0

bump-dependency

03

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

Bump a pinned dependency (TransformerEngine, Megatron-LM, NRX, etc.), regenerate the lockfile, open a PR, and drive it to green by attaching a watchdog to the "CICD NeMo" workflow and quarantining failing functional tests as flaky until the run is green.

not rated 893 +3 2d ago A 66 tokens original Apache-2.0

cicd

04

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

CI/CD reference for Megatron Bridge — pipeline structure, commit and PR workflow, CI failure investigation, and common failure patterns.

not rated 893 +3 2d ago C 29 tokens original Apache-2.0

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

Create or update concise, agent-readable Megatron Bridge model verification cards. Use when adding a model support card, auditing cross-model convergence comparability or verification coverage, recording conversion, deterministic inference, training, checkpoint resume, post-SFT export, performance, or weak-scaling…

not rated 893 +3 2d ago A 120 tokens original Apache-2.0

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

Code style and quality rules for Megatron Bridge — ruff configuration, naming conventions, type hints, mypy rules, docstrings, copyright headers, logging, and the code review checklist.

not rated 893 +3 2d ago A 44 tokens original Apache-2.0

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

Analyze and compare PyTorch CUDA memory snapshots produced by Megatron Bridge's ProfilingConfig(recordmemoryhistory=True). Replays the recorded allocation timeline to plot memory over time, compare two runs, and attribute peak memory to source code locations.

not rated 893 +3 2d ago A 56 tokens original Apache-2.0

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

Run Megatron-LM (MLM) and Megatron Bridge training with mock or real data. Covers correlation testing, available recipes, and multi-GPU examples.

not rated 893 +3 2d ago C 44 tokens original Apache-2.0

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

Convert single-node scripts to multi-node Slurm sbatch jobs and debug common multi-node failures. Covers srun-native vs uv run torch.distributed approaches, container setup, NCCL timeouts, OOM sizing for MoE models, and interactive allocation.

not rated 893 +3 2d ago C 61 tokens original Apache-2.0

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

Validate and use selective and full activation recompute in Megatron Bridge to reduce GPU memory usage at the cost of extra compute. Use for activation memory OOMs or regressions involving recomputegranularity, recomputenumlayers, recomputemodules, recomputemethod, selective recompute, full recompute, or activation…

not rated 893 +3 2d ago A 79 tokens original Apache-2.0

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

Validate and use CPU offloading in Megatron Bridge, including layer-level activation offloading and fractional optimizer state offloading with HybridDeviceOptimizer.

not rated 893 +3 2d ago A 39 tokens original Apache-2.0

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

Profile, validate, and use CUDA graph capture in Megatron Bridge, including local full-iteration graphs and Transformer Engine scoped graphs. Covers replay evidence, matched eager A/Bs, model-specific scopes, regressions, and failures.

not rated 893 +3 2d ago C 57 tokens original Apache-2.0

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

Validate and use MoE expert-parallel communication overlap in Megatron-Bridge, including overlapmoeexpertparallelcomm, delaywgradcompute, and flex dispatcher backends such as DeepEP and HybridEP.

not rated 893 +3 2d ago A 56 tokens original Apache-2.0

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

Operational guide for enabling Megatron FSDP in Megatron-Bridge, including config knobs, code anchors, pitfalls, and verification.

not rated 893 +3 2d ago A 40 tokens original Apache-2.0

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

Techniques for reducing peak GPU memory in Megatron Bridge, including expandable segments, PEFT plus sequence-parallel input re-gather, parallelism resizing, activation recompute, CPU offloading constraints, and common OOM fixes. Use for GPU OOMs, inadequate memory headroom, LoRA or PEFT activation pressure, memory…

not rated 893 +3 2d ago A 84 tokens original Apache-2.0

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

MoE expert-parallel communication overlap in Megatron Bridge. Covers dispatch/combine overlap, flex dispatcher backends, and expert wgrad scheduling.

not rated 893 +3 2d ago A 42 tokens original Apache-2.0

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

Select and validate an MoE token dispatcher (alltoall, DeepEP, or HybridEP) for a fixed workload and runtime. Covers backend availability, topology, matched A/B evidence, routing semantics, and failure diagnosis. Use when choosing a dispatcher or tracing a regression or crash to the MoE dispatcher configuration.

not rated 893 +3 2d ago A 76 tokens original Apache-2.0

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

Representative, point-in-time MoE training playbooks by hardware and model family. Use them as candidate seeds, then revalidate the exact runtime, semantics, topology, and steady-state throughput.

not rated 893 +3 2d ago A 51 tokens original Apache-2.0

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

Long-context MoE training guidance for Megatron Bridge. Covers CP sizing, selective recompute, dispatcher choices, and practical patterns from DSV3, Qwen3, and Qwen3-Next long-context experiments.

not rated 893 +3 2d ago A 55 tokens original Apache-2.0

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

Evidence-gated workflow for MoE performance optimization in Megatron Bridge. Covers measurement contracts, the Three Walls framework, parallel folding, profiling, matched A/B tuning, and final validation. Use for full MoE throughput tuning or diagnosing a regression involving memory, communication, compute, or…

not rated 893 +3 2d ago A 74 tokens original Apache-2.0

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

Practical guidance for training MoE VLMs in Megatron Bridge. Compares FSDP and 3D-parallel approaches, using rounded lessons from Qwen3-VL, Qwen3-Next, and other multimodal experiments.

not rated 893 +3 2d ago A 62 tokens original Apache-2.0

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

Analyze NVIDIA Nsight Systems .nsys-rep and exported .sqlite traces for Megatron Bridge training. Use for single-trace diagnosis, before/after comparisons, multi-rank surveys, slow-rank and pipeline-stage analysis, MFU or step-time investigations, GPU busy/compute-absent accounting, communication overlap and…

not rated 893 +3 2d ago A 118 tokens original Apache-2.0

NVIDIA-NeMo/Megatron-Bridge

Skill Claude CodeCodex

Operational guide for choosing and combining parallelism strategies in Megatron Bridge, including sizing rules, hardware topology mapping, and combined parallelism configuration.

not rated 893 +3 2d ago A 41 tokens original Apache-2.0

At most 3 mods per repository are shown here, and a mod shipped inside a plugin is left to that plugin's page — the rest are on their repository pages: