热点事件持续更新

CutBCE：JAX TPU 大词表 BCE 训练消除 OOM

1 篇报道1 个报道来源5 小时前更新

先了解这件事

AI 综述

CutBCE 是面向工业级推荐的精确大词表 BCE 损失与梯度，用 JAX + Pallas 实现，可覆盖 10^5–10^7 规模商品目录，目标是消除推荐 BCE 训练中的 OOM。2026-10-06 的新报道给出基准结果：单芯片 TPU v5e/v6 小基准上最高提速 91.9%；8 芯片 TPU 训练 87.6 万标签的多标签任务（Yarn-50M）时，峰值 HBM 下降 65.7%。早先报道只提到 JAX TPU 大词表训练提速 91.9%，后续报道把该提速定位到单芯片小基准，并补充了消除 OOM 与峰值 HBM 降 65.7% 的细节，两处 91.9% 的说法一致，未出现相互矛盾的数字。

AI 根据报道生成 · 3 小时前更新

报道时间线

沿着报道，了解事件的不同侧面。

10月6日

15:00
AGI Hunt
CutBCE：Google JAX TPU 核消除大规模推荐 BCE 训练 OOM，提速 91.9%
CutBCE 是面向工业级序列推荐的精确、硬件加速大词表 BCE 损失与梯度算子，用 JAX + Pallas 实现，可覆盖 10^5–10^7 规模商品目录。单芯片 TPU v5e/v6e 小基准上消除 OOM，最高提速 91.9%。8 芯片 TPU 训练 87.6 万商品的多标签 SASRec（Yambda-50M）时，峰值 HBM 降低 65.7%。

本事件热度走势

当前热度 9·可比范围峰值 9（10月6日 17:00）·近 24 小时可比范围变化 –

趋势仅比较持续完整观测到的相同主体，范围可能小于当前热度统计。移动指针或点击图表查看每小时热度；键盘可用左右方向键切换。