Skip to content

feat(mpo): add Nemotron Omni Nano and Super training#3345

Draft
jinglinglingling wants to merge 4 commits into
aroshanghias/nemotron-omni-basefrom
linglinj/nemotron-omni-mpo
Draft

feat(mpo): add Nemotron Omni Nano and Super training#3345
jinglinglingling wants to merge 4 commits into
aroshanghias/nemotron-omni-basefrom
linglinj/nemotron-omni-mpo

Conversation

@jinglinglingling

@jinglinglingling jinglinglingling commented Jul 25, 2026

Copy link
Copy Markdown
Contributor

Summary

  • add MPO with driver-owned reward-shift state and pair-safe preference packing
  • add MMPR multimodal preprocessing and Nano/Super recipes
  • add chained checkpoint and W&B resume support for time-limited Slurm jobs

Validation results

Nano CP1

Both packed and no-pack runs completed successfully. Their loss, reward, and accuracy trends are consistent with the legacy implementation.

Super CP2 (packed)

Loss and accuracy match the legacy trend through step 58. The increasing chosen BCO reward and decreasing rejected BCO reward show the expected preference separation.

The previous post-step-58 segment is excluded because it resumed without optimizer state. The corrected chained-resume smoke successfully restored model, distributed optimizer, LR scheduler, MPO state, and W&B run identity. A fresh chained 100-step run is now in progress.

Test plan

  • Run targeted MPO loss, data, packing, and recipe unit tests
  • Validate Nano CP1 packed and no-pack trends against legacy runs
  • Validate Super packed CP2 loss/accuracy through step 58
  • Verify optimizer/scheduler restoration across Slurm segments
  • Complete a continuous chained Super 100-step run
  • Send the dense decoder-mask fix as a dependent Megatron-Bridge change

Migrate MPO to model-owned multimodal packing with pair-safe batches, MMPR recipes, and resumable Nano/Super qualification workflows.
@copy-pr-bot

copy-pr-bot Bot commented Jul 25, 2026

Copy link
Copy Markdown

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@github-actions github-actions Bot added the Documentation Improvements or additions to documentation label Jul 25, 2026
@jinglinglingling jinglinglingling added the CI:Lfast Runs a fast test suite and re-use nightly `main` container (but sync dependencies to PRs version) label Jul 25, 2026
Allow smoke segments to keep one scheduler horizon while forcing an early checkpoint for resume validation.
@jinglinglingling

Copy link
Copy Markdown
Contributor Author

/ok to test efe1b3b

Linglin Jing added 2 commits July 24, 2026 22:06
Prefix chained-run IDs with letters so OmegaConf cannot parse hexadecimal IDs as scientific notation.
Keep the Python ID generator valid inside the shell command substitution.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

CI:Lfast Runs a fast test suite and re-use nightly `main` container (but sync dependencies to PRs version) Documentation Improvements or additions to documentation

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant