Skip to content

[PyTorch] Grouped MLP: emit FP32 scale gradient with no cast to BF16 - #3870

Open
sraman-rgb wants to merge 11 commits into
NVIDIA:mainfrom
sraman-rgb:sraman/grouped-mlp-fp32-scales-grad
Open

[PyTorch] Grouped MLP: emit FP32 scale gradient with no cast to BF16#3870
sraman-rgb wants to merge 11 commits into
NVIDIA:mainfrom
sraman-rgb:sraman/grouped-mlp-fp32-scales-grad

Revert properly

4e11b5e
Select commit
Loading
Failed to load commit list.
DCO / DCO succeeded Nov 14, 2026 in 0s

DCO

All commits are signed off!

View more details on DCO