JAXBench：面向自主 TPU 内核优化的基准测试
英文标题：JAXBench: Benchmarking Autonomous TPU Kernel Optimization
论文发布了 JAXBench，一套面向 TPU 的原生基准测试套件，用于评测 AI 生成的内核操作优化能力。作者指出，严格的基准测试已为自主 GPU 内核性能优化提供了共同的迭代目标，但 TPU 领域此前缺少对应标准。JAXBench 有望为开发者和研究人员提供统一评测目标，推动面向 TPU 的自动化内核优化方法进行可比较的性能改进。
打开原文相关 1.6 · 机会 0
2.1 KB
640 B
612 B
380 B
9.1 KB
32.7 KB
22.8 KB
24.2 KB
34.7 KB
中文
JAXBench：面向自主 TPU 内核优化的基准测试
arXiv:2607.20466v1
公告类型：新发布
摘要
严格的基准测试通过建立可持续优化的共同目标，推动了自主 GPU 内核性能优化的发展；但 TPU 领域尚不存在同等的基准。我们提出 JAXBench：一套 TPU 原生基准测试套件，用于评测 AI 生成的内核操作……
BibTeX 格式引用
English / 原文
BibTeX formatted citation
×