TPU kernel 很高 已验证

MaxKernel

一套多 agent 系统,用来生成和调优高性能 TPU kernel。实验用的是 Gemini 3.1 Pro,代码放在 Google 的 accelerator-agents 仓库。

论文写了三种做法:有人参与、完全自动、按图搜索。规划、实现、调试、测试、剖析这几类子 agent 是共用的。JaxBench 和真实模型负载上的实验用 Gemini 3.1 Pro。代码在 AI-Hypercomputer/accelerator-agents 的 maxkernel 目录。

人工参与

人定方向,agent 写代码. 系统里既有人参与的 agent,也有自动循环。论文是研究记录。仓库里有多少 kernel 是手写的,这里没有数。

证据

  • 论文
    All large language model queries and agent interactions throughout our experiments were conducted using the Gemini 3.1 Pro model. Code: https://github.com/AI-Hypercomputer/accelerator-agents/tree/main/MaxKernel
  • 官方仓库

备注

论文链接写的是 MaxKernel,GitHub 上能打开的路径是小写 maxkernel。目录里有 adk 和 claude-code 的辅助文件夹。论文里拿来评估的模型是 Gemini 3.1 Pro。