perf(recipe): align Nemotron 3 Nano library recipes - #4869
Draft
cuichenx wants to merge 16 commits into
Draft
Conversation
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>
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>
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>
Contributor
Author
|
/ok to test c410b86 |
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 does this PR do?
Align the public Nemotron 3 Nano library recipes with their validated H100 and
GB200 execution contracts while preserving convergence-sensitive training
semantics.
This current-
mainreconstruction:and PEFT wrappers plus package/top-level discovery exports;
and discovery tests;
disabled.
The library recipes do not clone benchmark recipes wholesale. Optimizer,
schedule, routing, loss, data/sample, initialization, and BF16 contracts remain
library-owned. Only evidenced execution settings are aligned.
Recipe mapping
moe,layernormrecompute, eager execution, TP overlap off, unfused native vocab-parallel CE, post-step cache release for checkpoint-resume headroom.Generic Nemotron 3.5 recipes introduced on current
mainremain independentof these specialized Nemotron 3 wrappers. In particular, generic Nemotron 3.5
SFT retains TP1/SP off, while its dedicated packed H100 recipe applies its own
topology.
The exact name
nemotron_3_nano_pretrain_8gpu_gb200_bf16_configis exported byboth library and performance packages. Unified runner lookup follows current
mainpolicy and selects the benchmark definition; the validated libraryrelease workload remains uniquely available as
nemotron_3_nano_gb200_pretrain_config.Convergence-sensitive exclusions
expert-bias behavior, token dropping, capacity, optimizer, LR schedule,
initialization, dropout, loss scaling/objective, or training samples.
removed because merged fix(ckpt): reduce distributed optimizer resume memory #4936 supersedes it.
main's pinned revision.Validation
Final head:
c410b864f53676613872e669de3ce9cbd75c4ed5, reconstructed onmain@81c47141f9507eb0a96069f302d54f11a148fa60with signed/DCO,ancestry-preserving merge commits. The original public head remains an
ancestor; publication was a normal fast-forward with no rebase, force-push, or
branch rename.
Mandatory native hardware gates
14300432--14300436, recovered from the preserved step-60 checkpoint by14367322--143673250.09788162, grad norm0.261; complete step-100 checkpoint.14300430COMPLETED 0:0.14300431COMPLETED 0:0.2476997, valid segment 1COMPLETED 0:0.2476998, valid segment 1COMPLETED 0:0.The H100 pretraining step-72 checkpoint write initially hit a transient
filesystem/PyTorch
unexpected posfailure after finite optimizer work. Theauthoritative pointer remained at complete step 60, the partial checkpoint was
preserved for evidence, and the minimal recovery chain loaded exact boundaries
and completed through step 100 without recurrence.
Current-main and local verification
905 passed.
pre-commit run --all-files: all hooks passed without changes.git diff --check, unmerged-entry, conflict-marker, exact-tree, DCO, andsignature checks passed.
generic Nemotron 3.5 SFT topology inheritance. Both are fixed and covered;
the exact final index had no remaining actionable findings.
Scope relative to merged work
and supplies the still-missing H100 alignment, GB200 SFT/PEFT, neutral
builder, and parity/denied-field coverage.
those changes are not present here.
Before this PR is ready for review
mainand resolved conflicts.