Jon Barron:JAX、PyTorch 等所有 softmax 实现都有两个隐蔽 bug

khademinori · x · 2026-10-24

Google 研究员 Jon Barron 发布新结果:包括 JAX、PyTorch 乃至教科书在内的所有现有 softmax 与 cross-entropy 实现都存在两个微妙的 bug。他提出一个简单的重写方式可同时修复两个问题,并能消除 grokking 现象中的 "slingshots"(突然跳变)。BlackHC 等研究者转发表示惊讶:ML 基础运算竟仍有如此大的改进空间。

所属事件:Barron 揭 softmax 与交叉熵实现两大数值 bug(21 条相关)→

原文链接 →

「研究」频道最新

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