Repository navigation
[NeMo FW 26.08 Release] MBridge 0.6.0 Roadmap #5144
Copy link
Copy link
Open
Labels
area:miscCross-cutting utilities, logging, helpers, and other changesCross-cutting utilities, logging, helpers, and other changesfeatureNew capabilities, enhancements, or enablement workNew capabilities, enhancements, or enablement workroadmaptrackingTracking issue for an ongoing project with smaller stepsTracking issue for an ongoing project with smaller steps
Description
Megatron Bridge — 26.08 Release Plan (Public)
Code freeze: ~Aug 1, 2026
This is a community-facing snapshot of the work targeted for the 26.08 release. Only items with a tracking PR or issue in NVIDIA-NeMo/Megatron-Bridge are listed. Larger multi-PR efforts (DeepSeek V4, MegatronMIMO, the MCore HybridModel migration, etc.) are tracked in their own issues — progress is updated there, and those issues are linked below rather than duplicated here.
1. Model Support
LLM
| Model | Status | Link |
|---|---|---|
| Qwen3.5 text bridges (dense + MoE) | Merged | PR #3769 (community @HowardZorn) |
| GLM-5.2 (text LLM; GB200/H100 recipes, cuDNN DSA path) | Merged | PR #4520 (community @HollowMan6), issue #4490 |
| EXAONE 4.0 (LG AI Research) | Merged | PR #2532 (community @Bias92) |
| Gemma-4 E4B | Merged | PR #4148 (community @DOGEUNNKIM) |
| HunYuan V3 (MoE causal LM) | Merged | PR #4680 (community @xhx1022), AutoBridge registration PR #4822 |
| K-EXAONE (EXAONE MoE) | Merged | PR #4730 (community @sangHa0411) |
| Nemotron 3.5 | Merged | PR #5035 |
VLM / Multimodal
| Model | Status | Link |
|---|---|---|
| EXAONE 4.5 (EXAONE 4.0 32B base + vision encoder) | Merged | PR #4730 (community @sangHa0411) |
| MiniMax M3 (full multimodal — text + vision) | Merged | LM bridge PR #4652 (community @khazic); VLM import/export PR #5058; issue #4585 |
Diffusion / Masked-Diffusion
| Model | Status | Link |
|---|---|---|
| LLaDA 1.5 (masked-diffusion LM bridge) | Merged | PR #4019 |
Tracked separately (progress updated in the linked issue)
- DeepSeek V4 — enhanced support & verification — issue PoR: Deepseek v4 Roadmap #4633
- Model framework: migrate to MCore
HybridModel(cross-model refactor) — issue PoR: migrate to MCore HybridModel #4510 - Model support verification levels (full-model conversion + inference validation) — issue POR: Track model support verification levels #4509
2. Training Features
Merged
- Unified dataset configuration, processing, collation & packing architecture — shared Config + Builder paths across text and multimodal datasets, common masking/padding/packing contracts, model-owned VLM collators, direct THD packing — PR refactor(data): adopt config and builder dataset architecture #4685, PR refactor(data)!: unify multimodal dataset builders #4774, PR [data] fix: isolate and unify Nemotron Omni collation #4776
- Masked-LM HF-pretrained wrapper support — AutoBridge selects
PreTrainedMaskedLMfor*ForMaskedLMarchitectures — PR feat(model): add masked lm support. #4735 (community @bbuschkaemper) - Checkpoint, resume & export usability — save HF weights alongside native Megatron checkpoints (PR [training, ckpt] feat: Save HF weights alongside native Megatron checkpoints #3825, community @dancingpipi); Energon dataloader save/resume (PR [ckpt, data] feat: complete Energon dataloader checkpointing save/res… #4521, community @zsombor-kaiko); distributed-optimizer resume memory fix (PR fix(ckpt): reduce distributed optimizer resume memory #4936)
- PEFT / LoRA scaling — distributed adapter export (PR feat(conversion): support distributed adapter export #4221); EP-reshardable per-expert LoRA (PR Changes to support EP reshardable per-expert LoRA #3951); sequence-parallel LoRA activation-memory reduction (PR [PEFT] feat: reduce LoRA sequence-parallel activation memory #4581); shared outer LoRA (PR Add shared outer lora #3958, community @nanjiangwill)
- Fine-tuning & evaluation usability — epoch-based training duration (PR feat(training): support epoch-based training duration #4147, community @muyihao); averaged eval losses + pure-eval example (PR feat(training): return eval losses and add pure eval example #4125)
- Quantized HF export foundations (ModelOpt) — NVFP4 HF export path (PR Add ModelOpt NVFP4 HF weight export support #4102, community @HollowMan6); W4A16/W4A4 NVFP4 grouped-MoE export (PR support modelopt W4A16 & W4A4 NVFP4 export for grouped MoE weights #4566, community @HollowMan6); KV-cache amax mapping (PR feat(quant): Add modelopt KV cache amax mapping. #4591)
- VLM data pipeline refactor (processing / batching / packing / masking) — issue PoR: [VLM] Refactor: clean up VLM data pipeline (processing / batching / packing / masking) #4041
Tracked separately (progress updated in the linked issue)
- MegatronMIMO 2026 Q3–Q4 roadmap (DistTrain + MIMO extensions) — issue PoR: MegatronMIMO 2026 Q3-Q4 Roadmap #4392
- Dynamic (Hybrid) Context Parallelism — issue [support] Dynamic (Hybrid) Context Parallelism: enablement status and end-to-end verification in Megatron-Bridge #4586
- Training-loop upstreaming to Megatron Core — issue PoR: align Megatron-LM and Megatron-Bridge checkpoints #4595
- SFT dataset unification / upstreaming with Megatron Core — issue [tracking] SFT dataset unification / upstreaming with Megatron Core #4596
- Merge MLM dev-branch MoE recipes into Bridge recipes — issue PoR: Merge MLM dev branch MoE recipes into Bridge training recipes #4471
3. Performance & Recipes
Merged (foundation)
- Unified flat performance recipes for all model families — PR [recipe][perf] feat: unified flat performance recipes for all model families #2803; H100 library recipe namespace (PR refactor(recipes): add h100 recipe namespace #4622); legacy perf-config removal (PR refactor(performance): remove legacy perf configs #4623); recipe environment-variable support (PR feat(recipes): support recipe environment variables #4682); perf recipes routed by name through
train.sh --recipe(PR feat(training): route performance recipes by name #4945)
Tracked separately (progress updated in the linked issue)
- Merge performance recipes and model recipes with convergence validation — issue PoR: Merge performance recipes and model recipes with convergence validation #4582
- Megatron FSDP in Qwen3-VL 30B-A3B — issue PoR: Megatron FSDP in Qwen3-VL 30B-A3B #4514
4. UX & Infrastructure
Merged
- Unified training & checkpoint-conversion CLI entry points — supported
scripts/training/train.shlauncher (PR feat(training): add shared recipe launcher #4856, PR feat(training): route performance recipes by name #4945);scripts/conversion/convert.shand distributed roundtrip (PR feat(conversion): add shared checkpoint launcher #4873, PR feat(ckpt): add roundtrip conversion launcher #4886)
5. Examples & Tutorials
Merged
- Nemotron 3 Ultra recipes and examples — conversion, inference, DCLM pretraining, packed OpenMath SFT/PEFT, packing scripts/docs — PR feat(nemotronh): add Nemotron 3 Ultra recipes and examples #4224
- Qwen3.5-VL MegatronMIMO non-colocated examples — conversion, HF-data SFT, Slurm launch, tutorial docs — PR Add Qwen3.5-VL MegatronMIMO examples #4071, PR Add Qwen3.5-VL MegatronMIMO non-colocated SFT tutorial #4239
- MoE recipe notebook — PR [tutorial] feat: add MoE notebook example #3890 (community @karinseve)
Tracked separately (progress updated in the linked issue)
- VLM dataset benchmark / support (hi-res & interleaved multi-image) — issue [data] tracking: VLM SFT benchmark — high-res & interleaved multi-image dataset support #3133
- Extended Megatron FSDP example: verify combinations & document caveats — issue [26.08] Extended Megatron FSDP example: verify combinations & document caveats #4246
Reactions are currently unavailable
Metadata
Metadata
Assignees
Labels
area:miscCross-cutting utilities, logging, helpers, and other changesCross-cutting utilities, logging, helpers, and other changesfeatureNew capabilities, enhancements, or enablement workNew capabilities, enhancements, or enablement workroadmaptrackingTracking issue for an ongoing project with smaller stepsTracking issue for an ongoing project with smaller steps
Activity
[-][NeMo FW 26.08 Release] MBridge 26.08 Roadmap[/-][+][NeMo FW 26.08 Release] MBridge 0.6.0 Roadmap[/+]