fix(loss): compute causal-LM cross-entropy in float32 - #464

Merged

le1nux merged 2 commits into

mainfrom

fix/cross-entropy-fp32-upcast

Sep 14, 2026

Commits on Sep 14, 2026

fix(loss): compute causal-LM cross-entropy in float32

le1nux
and
claude
committed
64f661a
fix(loss): correct the accuracy test's reference and the precision rationale

le1nux
and
claude
committed
70cbb83