研究人员开发了 MaxKernel,一个旨在自动化为 Tensor Processing Units (TPU) 创建高性能定制内核的多智能体系统。该系统采用三种不同的方法:用于协作设计的“人在回路”智能体、用于自动化优化的“自主”智能体以及用于更广泛探索的“基于图的自主搜索”。MaxKernel 利用专门的子智能体来执行规划、调试和测试等任务,并已证明其能够生成在 JaxBench 等基准测试中达到专家级性能的优化实现。 AI
影响 这项研究可能会显著加快 AI 硬件优化代码的开发速度,从而可能降低复杂加速器编程的入门门槛。
排序理由 该集群描述了一篇研究论文,其中详细介绍了一种使用 AI 智能体生成 TPU 内核的新方法。
- JAXBench
- large-language models
- MaxKernel
- accelerator-agents
- AI-Hypercomputer
- Catalonia
- Graph-Based Autonomous Search
- interactive machine learning
AI 生成摘要 · Google Gemini · 来自 2 个来源。 我们如何撰写摘要 →