[do not merge] backport Qwen3.5 and Nemotron 3 Nano pretraining to r0.5.0 - #4871
Closed
cuichenx wants to merge 8 commits into
Closed
[do not merge] backport Qwen3.5 and Nemotron 3 Nano pretraining to r0.5.0#4871cuichenx wants to merge 8 commits into
cuichenx wants to merge 8 commits into
Conversation
Signed-off-by: Chen Cui <chcui@nvidia.com>
Signed-off-by: Chen Cui <chcui@nvidia.com>
Signed-off-by: Chen Cui <chcui@nvidia.com>
|
Auto-sync is disabled for draft pull requests in this repository. Workflows must be run manually. Contributors can view more details about this message here. |
Signed-off-by: Chen Cui <chcui@nvidia.com>
Signed-off-by: Chen Cui <chcui@nvidia.com>
Signed-off-by: Chen Cui <chcui@nvidia.com>
Signed-off-by: Chen Cui <chcui@nvidia.com>
Signed-off-by: Chen Cui <chcui@nvidia.com>
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
What changed
Backports the Qwen3.5 text-only and Nemotron 3 Nano GB200 library pretraining recipes onto
r0.5.0for NeMo 26.06 convergence validation.Qwen3.5
qwen35_9b_pretrain_configandqwen35_35b_a3b_pretrain_configaliasesNemotron 3 Nano
nemotron_3_nano_pretrain_8gpu_gb200_bf16_configand the NeMo-CI selector aliasnemotron_3_nano_gb200_pretrain_configmoe_hybridep_num_sms=16NeMo-CI should select this recipe with
MODEL_RECIPE_NAME=nemotron_3_nano_gb200. The shared release template then deliberately overrides only the convergence-run controls to 48,000 steps, sequence length 4096, and GBS512; the model case overrides GPU count to 64 and MBS to 2.Why
NeMo-CI needs a reviewable release-branch SHA for the paired Qwen3.5 9B and Nemotron 3 Nano 100B-token, 64-GPU DGX Cloud GB200 convergence candidates against the 26.06 container. This does not replace the corresponding main-targeting recipe work.
Validation
nvcr.io/nvidian/nemo:26.06.01.rc2with release-pinned Megatron-Core458c8d0escripts/performance/run_recipe.pydry-run resolvednemotron_3_nano_gb200with the exact 64-GPU NeMo-CI overlay: GBS512, MBS2, sequence length 4096, and 48,000 stepsgit diff --checkpassed2382796: 100/100, finite loss and grad norm, zero skipped/NaN iterations, checkpoint and validation/test complete2378268, W&Bhvdq6oa3Active convergence run
The corrected Qwen-only NeMo-CI overlay root
58144288is pinned to Qwen headdd81e664eand is unaffected by this Nemotron addition. A Nemotron launch must use current PR head987244934andMODEL_RECIPE_NAME=nemotron_3_nano_gb200.