Jon Barron:JAX、PyTorch 的 softmax 与交叉熵实现都藏着两个数值 bug

jon_barron · x · 2026-10-24

Google 研究员 Jon Barron 宣布新结果:现有所有 softmax 和交叉熵实现(包括 JAX、PyTorch 及教科书写法)都存在两个微妙的数值 bug,一个简单的重写即可修复,并能消除 grokking 过程中的「slingshot」突变现象。

所属事件:谷歌研究员称 JAX/PyTorch 的 Softmax 与交叉熵实现全有 bug(6 条相关)→

原文链接 →

「研究」频道最新

更多「研究」频道 AI 资讯 →