AI-Hypercomputer

MaxKernel 使用多智能体 AI 自动生成 TPU 内核

研究人员开发了 MaxKernel，一个旨在自动化为 Tensor Processing Units (TPU) 创建高性能定制内核的多智能体系统。该系统采用三种不同的方法：用于协作设计的“人在回路”智能体、用于自动化优化的“自主”智能体以及用于更广泛探索的“基于图的自主搜索”。MaxKernel 利用专门的子智能体来执行规划、调试和测试等任务，并已证明其能够生成在 JaxBench 等基准测试中达到专家级性能的优化实现。