Fixing Softmax and Cross-Entropy: Two Numerical Pitfalls

Softmax Collapse and Softmax Cancellation: two floating-point pitfalls in Softmax and Cross-Entropy, and mathematically equivalent fixes.

October 25, 2026 · 3 min · liqq