发生了什么
Google, Harvard, UC Berkeley 发表 JAXBench(50个JAX工作负载)
为什么值得关注
首次提出面向TPU的JAX内核优化基准JAXBench，并展示了文档条件化带来的显著提升，值得TPU优化领域从业者阅读原文。
信息来源
以下内容来自公开来源，可打开原文继续核验。
01
JAXBench与Pallas内核优化
Google、哈佛和UC Berkeley联合发布JAXBench，包含50个基于真实MaxText架构的JAX工作负载。使用Gemini 3 Flash并配合文档条件化，将正确率从5.8%提升至37.3%，解决48/50个基准，速度提升1.28倍，波束搜索进一步优化至1.36倍。
打开原始来源 ↗