Skip to content

fix(losses): avoid NaN gradient in weighted_logsoftmax at -inf logits - #1762

Open
jaideeppyne wants to merge 1 commit into
google-deepmind:mainfrom
jaideeppyne:fix/weighted-logsoftmax-jvp-nan
Open

fix(losses): avoid NaN gradient in weighted_logsoftmax at -inf logits#1762
jaideeppyne wants to merge 1 commit into
google-deepmind:mainfrom
jaideeppyne:fix/weighted-logsoftmax-jvp-nan

Commits

  1. Commits on Aug 31, 2026