[pull] main from NVIDIA:main#322

Merged

pull[bot] merged 2 commits intoVictarry:mainfrom

Jul 9, 2025

Commits on Jul 9, 2025

[PyTorch] Tests for loading previously-generated checkpoints (#1899)

timmoon10
and
ksivaman
authored
[JAX] Fix grouped GEMM error on CUDA 12.9.1 & later (#1925)
huanghua1994
authored