MaxKernel:面向 TPU 的智能体化内核生成系统
MaxKernel: Agentic Kernel Generation for TPUs
阅读原文
本站未展示全文,请前往来源网站阅读。
AI 导读
MaxKernel 是一个面向 TPU 内核开发的多智能体系统,提供 Human-in-the-Loop、全自动 Auto 以及基于图的自主搜索三种范式,共享规划、实现、自调试、测试与硬件剖析等子智能体。该系统在包含 50 项内核任务的 JaxBench 及多个开源模型真实负载上评测,生成的实现可匹配专家手工调优基线。MaxKernel 已开源。
来源:Hugging Face Daily Papers · arxiv.org