Repository navigation

MxFP8 gradient cross entropy option - #3758

Open

sraman-rgb wants to merge 9 commits into

NVIDIA:mainfrom

sraman-rgb:sraman/te-ce-mxfp8-grad

Open

MxFP8 gradient cross entropy option#3758
sraman-rgb wants to merge 9 commits into
NVIDIA:mainfrom
sraman-rgb:sraman/te-ce-mxfp8-grad

Commits on Oct 27, 2026
- [PyTorch] Cross entropy: MXFP8 gradient output and faster single-rank launch
  
  sraman-rgb
  and
  claude
  committed
  d9b0681
- [PyTorch] Cross entropy: drop non-finite special cases from the MXFP8 scale
  
  sraman-rgb
  and
  claude
  committed
  4c6363e
Commits on Nov 10, 2026
- Merge branch 'main' into sraman/te-ce-mxfp8-grad
  ksivaman
  authored
  88d8a47
Commits on Nov 11, 2026
- Merge branch 'main' into sraman/te-ce-mxfp8-grad
  ksivaman
  authored
  3e251a9
- [PyTorch] Cross entropy: keep masked logits from turning the softmax sum into NaN
  
  sraman-rgb
  and
  claude
  committed
  e41b2be
Commits on Nov 13, 2026
- NaN
  ksivaman
  committed
  a41ae4b
- Merge branch 'main' into sraman/te-ce-mxfp8-grad
  ksivaman
  authored
  5bc1e72
- Merge branch 'ksivamani/nan_fix' into sraman/te-ce-mxfp8-grad
  
  ksivaman
  and
  claude
  committed
  bf6b1b0
- Small testing refactors
  ksivaman
  committed
  c64df22