Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
42 commits
Select commit Hold shift + click to select a range
5e31514
Create a Protocol for the MLP layer of TransformerLayer (#3435)
nschank May 10, 2026
a2ec5c1
Revert "Add Python-side guardrail for HybridEP InfiniBand limit and r…
ko3n1g May 11, 2026
e93755e
chore(beep boop 🤖): Bump (main) (2026-05-11)
github-actions[bot] May 11, 2026
ad58411
Add Python-side guardrail for DeepEP IB limits (#4719)
janEbert May 11, 2026
5123f6a
ci: revert bad uv.lock bump and label future bumps with `Run function…
ko3n1g May 11, 2026
33d47e0
[ci] fix: treat cancelled run-main-script step as failure (#4727)
ko3n1g May 11, 2026
e42e2fa
ci: Major refactor of release-workflows (#4602)
ko3n1g May 11, 2026
434368c
build(deps): bump nvidia-modelopt to 0.43 (#4723)
ko3n1g May 11, 2026
74687fe
fix(fsdp): recognize legacy GDN TP metadata (#4664)
Glitchfix May 11, 2026
9718f7d
Fixes for Nemotron3 Super release test config (#4544)
maanug-nv May 11, 2026
97f3bce
feat(gpt): add output postprocess hook (#4686)
Glitchfix May 11, 2026
f744215
Add bump-base-image skill and update golden value comparison (#4733)
balasaajay May 11, 2026
6486d52
Guard omegaconf imports (#4685)
maanug-nv May 11, 2026
7d24b28
Fix a regression introduced by #4625 for nightly runs (#4734)
balasaajay May 12, 2026
6853b47
Add LLaVA audio (sound) model support (#4402)
cuichenx May 12, 2026
86bf476
Support transfomers 5.x.x for text generation server (#4732)
tdene May 12, 2026
815c83d
Update transformer-engine dependency to version 2.15.0 (#4682)
balasaajay May 12, 2026
72dd053
Increase CG cover from max_requests to max_tokens (#4214)
tdene May 13, 2026
f8c942b
chore: rotate oncall schedule
github-actions[bot] May 13, 2026
10b514b
fully remove legacy code (#4759)
dimapihtar May 13, 2026
fc41581
fix legacy torch save when tensor_model_parallel_size > expert_model_…
dimapihtar May 13, 2026
d802f09
Wire --rl-inference-parsers into MRL (#4768)
tdene May 13, 2026
c1e938b
Integrate LayerWiseDistributedOptimizer with DDP buffer infrastructur…
deepakn94 May 13, 2026
0dc36df
[training migration] Migrate mamba builder (#4550)
maanug-nv May 13, 2026
e35d4e5
NCCL UB fix: reduce memory cost and correctly deregister NCCL mem poo…
xrennvidia May 13, 2026
1ba0aa9
fix: use no_mask in local ViT layer spec (#4395)
Phlip79 May 14, 2026
20bf831
refit clean up and refactoring (#4762)
wdykas May 14, 2026
a7c9e8c
Make weight and optimizer memory estimation take into account expert …
YangFei1990 May 14, 2026
118933a
Support recomputing in HybridModel (#4496)
xuantengh May 14, 2026
925422c
One single flag that determines if we are in inference (#4617)
tdene May 14, 2026
2d1fa8d
[main] feat(moe): Support packed sequence for gated delta net (GDN) (…
yuzhongw-nvidia May 14, 2026
58d45bf
remove dead manual_release_grads code path in 1F1B overlap schedule (…
Wohox May 14, 2026
ada8dfe
Fix recompute checkpointing + training CGs (#3919)
tdene May 14, 2026
dbfc96b
Use Protocols to type-check linear_proj submodules of Attention (#3434)
nschank May 14, 2026
d167123
fix tokenizers in respect to newer transformers (#4608)
dimapihtar May 14, 2026
98031e1
Bump nvidia-modelopt>=0.44.0 (#4803)
kevalmorabia97 May 14, 2026
266562f
Update owners (#4794)
Phlip79 May 14, 2026
e9a0930
ci: Update workflow to use same commit for building docker image and …
balasaajay May 14, 2026
a53107c
chore: Update nightly tests golden values (#4805)
balasaajay May 14, 2026
9b4074b
Inference: Optimize Prefill Engine Steps for Nemotron (#4764)
sidsingh-nvidia May 14, 2026
52c54ec
Merge remote-tracking branch 'origin/main' into main2dev/15_05_2026
svcnvidia-nemo-ci May 15, 2026
6d3abeb
fix: post-CI corrections
svcnvidia-nemo-ci May 15, 2026
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
10 changes: 8 additions & 2 deletions .github/actions/action.yml
Original file line number Diff line number Diff line change
Expand Up @@ -209,6 +209,8 @@ runs:
if: always()
env:
IS_UNIT_TEST: ${{ inputs.is_unit_test == 'true' }}
MAIN_CONCLUSION: ${{ steps.run-main-script.conclusion }}
MAIN_EXIT_CODE: ${{ steps.run-main-script.outputs.exit_code }}
run: |
logs_report=logs-${{ inputs.test_case }}-${{ github.run_id }}-$(cat /proc/sys/kernel/random/uuid)
echo "logs_report=$logs_report" | sed 's/\//-/g' | sed 's/\*/-/g' | tee -a "$GITHUB_OUTPUT"
Expand All @@ -219,8 +221,12 @@ runs:
fi
echo "coverage_report=$coverage_report" | tee -a "$GITHUB_OUTPUT"

EXIT_CODE=${{ steps.run-main-script.outputs.exit_code }}
IS_SUCCESS=$([[ "$EXIT_CODE" -eq 0 ]] && echo "true" || echo "false")
EXIT_CODE="${MAIN_EXIT_CODE:-${MAIN_CONCLUSION}}"
if [[ "$MAIN_CONCLUSION" == "success" ]]; then
IS_SUCCESS=true
else
IS_SUCCESS=false
fi

if [[ "$IS_SUCCESS" == "false" && "${{ inputs.is-optional }}" == "true" ]]; then
echo "::warning::Test failed but is marked optional — treating as success."
Expand Down
8 changes: 4 additions & 4 deletions .github/oncall_schedule.json
Original file line number Diff line number Diff line change
@@ -1,8 +1,4 @@
[
{
"user": "janEbert",
"date": "2026-05-06"
},
{
"user": "dimapihtar",
"date": "2026-05-13"
Expand Down Expand Up @@ -46,5 +42,9 @@
{
"user": "Phlip79",
"date": "2026-07-22"
},
{
"user": "YangFei1990",
"date": "2026-07-29"
}
]
26 changes: 17 additions & 9 deletions .github/workflows/_build_test_publish_wheel.yml
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@ on:
default: true
secrets:
TWINE_PASSWORD:
required: true
required: false

jobs:
build-and-test-wheels:
Expand Down Expand Up @@ -74,7 +74,7 @@ jobs:
rm LICENSE || true
docker run --rm -e NO_VCS_VERSION=1 -v $(pwd):/workspace -w /workspace $IMAGE bash -c '\
for python_version in cp311 cp312 cp313; do \
/opt/python/${python_version}-${python_version}/bin/pip install --upgrade "setuptools<80.0.0,>=77.0.0" build; \
/opt/python/${python_version}-${python_version}/bin/pip install --upgrade "setuptools>=80" build; \
done && \
for python_version in cp311 cp312 cp313; do \
/opt/python/${python_version}-${python_version}/bin/python -m build; \
Expand Down Expand Up @@ -144,7 +144,8 @@ jobs:
publish-wheels:
needs: [build-and-test-wheels]
runs-on: ubuntu-latest
if: inputs.no-publish == false
environment:
name: ${{ inputs.no-publish && 'public' || 'main' }}
strategy:
fail-fast: false
matrix:
Expand All @@ -171,6 +172,7 @@ jobs:
TWINE_PASSWORD: ${{ secrets.TWINE_PASSWORD }}
TWINE_REPOSITORY: ${{ (github.ref == 'refs/heads/main' || startsWith(github.ref, 'refs/heads/r')) && 'pypi' || 'testpypi' }}
PLATFORM: ${{ matrix.PLATFORM }}
DRY_RUN: ${{ inputs.no-publish }}
run: |

# Delete sdist for arm64 since we already upload it with amd64.
Expand All @@ -180,9 +182,15 @@ jobs:

ls -al dist/
pip install twine
twine upload \
--verbose \
-r $TWINE_REPOSITORY \
-u $TWINE_USERNAME \
-p $TWINE_PASSWORD \
dist/*

if [[ "$DRY_RUN" == "false" ]]; then
[[ -z "$TWINE_PASSWORD" ]] && { echo "::error::TWINE_PASSWORD unset"; exit 1; }
twine upload \
--verbose \
-r $TWINE_REPOSITORY \
-u $TWINE_USERNAME \
-p $TWINE_PASSWORD \
dist/*
else
echo "[dry-run] would execute: twine upload --verbose -r $TWINE_REPOSITORY -u <user> -p <pass> dist/*"
fi
Loading
Loading