Skip to content

cp: feat(recipe): add deterministic Nemotron 3 Ultra, Qwen3.5-9B and Moonlight-16B pretrain recipes (6440) into r0.7.0 - #6497

Merged
cuichenx merged 1 commit into
NVIDIA-NeMo:r0.7.0from
ZhiyuLi-Nvidia:zhiyul/cp-6272-6440-to-r0.7.0
Oct 16, 2026
Merged

cuichenx merged 1 commit into
NVIDIA-NeMo:r0.7.0from
ZhiyuLi-Nvidia:zhiyul/cp-6272-6440-to-r0.7.0

Conversation

@ZhiyuLi-Nvidia

@ZhiyuLi-Nvidia ZhiyuLi-Nvidia commented Oct 15, 2026 •

Copy link
Copy Markdown
Contributor

What does this PR do ?

Cherry-pick #6440 from main into r0.7.0: deterministic pretrain recipes for Nemotron 3 Ultra/Nano (GB200), Qwen3.5-9B (GB200) and Moonlight-16B (H100).

Changelog

Picked with git cherry-pick -x -s (one commit, 3b91885d5), rebased onto r0.7.0 at 39ab66835 (#6439):

  • Deterministic recipes for Nemotron 3 Ultra/Nano, Qwen3.5-9B and Moonlight-16B, with docs and unit tests.
  • Adds DETERMINISM_ENV_VARS in recipes/utils/environment_utils.py and a small change in training/initialize.py.
  • Switches the DSv3 GB300 deterministic recipe's env_vars to **DETERMINISM_ENV_VARS.

Conflict resolution (one, in perf_recipes/deepseek/gb300/deepseek_v3.py): r0.7.0 already had the explicit determinism env block from #6439. I replaced it with **DETERMINISM_ENV_VARS, as in #6440. The resulting function matches main.

Not run locally: unit tests (no working Python env on the login node); relying on CI.

GitHub Actions CI

See the CI section in the Contributing doc for how to trigger the CI. A Nvidia developer will need to approve and trigger the CI for external contributors.

Before your PR is "Ready for review"

Pre checks:

  • Make sure you read and followed Contributor guidelines
  • Did you write any new necessary tests? (tests come from the original PR)
  • Did you add or update any necessary documentation? (docs come from the original PR)
  • Does the PR affect components that are optional to install? (Ex: Numba, Pynini, Apex etc)
    • Reviewer: Does the PR have correct import guards for all optional libraries?

Additional Information

🤖 Generated with Claude Code

@copy-pr-bot

copy-pr-bot Bot commented Oct 15, 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.

cuichenx
cuichenx previously approved these changes Oct 15, 2026
…light-16B pretrain recipes (NVIDIA-NeMo#6440)

Signed-off-by: Zhiyu Li <zhiyul@nvidia.com>
Signed-off-by: adityavavreNVDA <avavre@nvidia.com>
Co-authored-by: Claude Sonnet 5.5 <noreply@anthropic.com>
Co-authored-by: adityavavreNVDA <avavre@nvidia.com>
(cherry picked from commit 7c2c9ba)
Signed-off-by: Zhiyu Li <zhiyul@nvidia.com>
@ZhiyuLi-Nvidia
ZhiyuLi-Nvidia force-pushed the zhiyul/cp-6272-6440-to-r0.7.0 branch from 07d636a to 3b91885 Compare October 16, 2026 00:05
@ZhiyuLi-Nvidia ZhiyuLi-Nvidia changed the title Cherrypick 6272 6440 to r0.7.0 cp: feat(recipe): add deterministic Nemotron 3 Ultra, Qwen3.5-9B and Moonlight-16B pretrain recipes (6440) into r0.7.0 Oct 16, 2026
@cuichenx cuichenx added feature New capabilities, enhancements, or enablement work area:recipe Training recipes and launch configs needs-review PR is ready for code review and waiting on a reviewer labels Oct 16, 2026
@cuichenx

Copy link
Copy Markdown
Contributor

/ok to test 3b91885

@github-actions

Copy link
Copy Markdown
Contributor

Automatic Claude reviews have been retired. To request a pull-request review, post a comment containing:

/review

Add model=claude to use a Claude reviewer (the default is model=codex). mode=light|strict selects the review depth; for example, /review model=claude mode=strict. Comment /review help for all options.

@github-actions

Copy link
Copy Markdown
Contributor

@cuichenx
cuichenx merged commit 057bdbd into NVIDIA-NeMo:r0.7.0 Oct 16, 2026
91 of 94 checks passed

This branch was successfully deployed

1 active deployment
test — 3b91885d Deployed Oct 16, 2026 by copy-pr-bot[bot] via cicd-wait-in-queue #22949
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

area:recipe Training recipes and launch configs feature New capabilities, enhancements, or enablement work needs-review PR is ready for code review and waiting on a reviewer

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants