MaxKernel：面向TPU的智能体式内核生成

MaxKernel用多智能体自动生成TPU内核，报告可匹配专家手工调优基线。

资讯摘要

论文提出MaxKernel，一个用于TPU内核开发的多智能体系统，包含人在回路、全自动指标/追踪驱动优化循环、基于图的自主搜索三种范式，共享规划、实现、自调试、测试与硬件剖析等子智能体。作者在含50项内核任务的JaxBench及开源模型真实负载上评估，报告其生成实现可匹配专家手工调优基线，并已开源。

要点速读

1. MaxKernel提供人在回路、Auto全自动优化循环、基于图的自主搜索三种TPU内核开发范式。
2. 系统共享规划、实现、自调试、测试和硬件剖析等专用子智能体。
3. 作者在JaxBench的50项TPU内核任务及开源模型真实负载上评估，报告可匹配专家手工调优基线，代码已开源。

中文摘要由 AI 根据公开来源生成，可能存在理解偏差；具体能力、数字、开放时间等信息请以原文为准。
采集于 2026/11/5 08:00:38（北京时间）。 arXiv 日期为本次公告时间，论文初次投稿日期请查看原文版本记录。