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 替代不了的工作何时更值钱 — soumitrashukla9 · 2026-10-28
- IEEE IT 刊文:希尔伯特空间高斯测度间的几何 Jensen-Shannon 散度 — FrnkNlsn · 2026-10-28
- CPR 数据集:现有检索器跨领域联想最佳 Recall@10 仅 0.26 — coallaoh · 2026-10-28
- HuggingFace 机器鸭雪地行走:29 万片雪花物理仿真有多较真 — lukas_m_ziegler · 2026-10-28
- Mercor 联合斯坦福、NBER 发布开源任务目录 TaskNet,补齐 O*NET 盲区 — soumitrashukla9 · 2026-10-28
- François Fleuret 提出对齐三步法:预训练加 RL、给 token 标「想/说」 — francoisfleuret · 2026-10-28