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

Commits

  1. Commits on Nov 9, 2026

  2. Commits on Nov 13, 2026

  3. Commits on Nov 14, 2026