跳到正文
arXiv · cs.AI · Shangkun Wang, Nina Cai, Charles Hoong, Julian Walker, Gerson Kroiz, George Vanica, Deepak Patil, Andi Gavrilescu, Hassan Sipra, Sethu Sankaran · 6 小时前模型评分42

MaxKernel:面向 张量处理器(TPU,Tensor Processing Unit) 的智能体内核生成系统

MaxKernel: Agentic Kernel Generation for TPUs

导读

MaxKernel 多智能体系统用于生成 张量处理器(TPU,Tensor Processing Unit) 高性能内核,提供人在回路、全自动和基于图的自主搜索三种模式,共享规划、实现、自调试、测试与硬件分析等子智能体。该系统在包含 50 项内核任务的 JaxBench 上评估,生成结果可匹配专家手工调优基线,并已开源。

分类置信度

中 · 待人工复核

  • 只有单一媒体来源,需要人工复核。

来源级别:媒体与个人

事件类型:研究发布

证据摘录

  • “In this work, we present MaxKernel, a multi-agent system that implements three distinct paradigms for TPU kernel development”
  • “Our agent is open-sourced and available this https URL .”

来源:arXiv · cs.AI · arxiv.org