Skip to content

[PyTorch] Fix distributed BF16 weights and trim NVFP4 saved state in the fused grouped MLP - #3545

Merged
ksivaman merged 8 commits into
NVIDIA:mainfrom
xrennvidia:xren/dist-weight-saved-tensor-hooks
Oct 26, 2026
Merged

[PyTorch] Fix distributed BF16 weights and trim NVFP4 saved state in the fused grouped MLP#3545
ksivaman merged 8 commits into
NVIDIA:mainfrom
xrennvidia:xren/dist-weight-saved-tensor-hooks

[PyTorch] Skip columnwise quantization of gathered distributed FC1 we…

274706e
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Oct 23, 2026 in 1s

DCO

All commits are signed off!

View more details on DCO