MaxKernel：面向 TPU 的智能体式内核生成
MaxKernel 是一个用于设计和优化 TPU 定制内核的多智能体系统。它支持三种方式：人类参与的协作式设计、基于性能指标和追踪数据的全自动优化，以及用于更广泛探索设计空间的图搜索式自主探索。多个专业子智能体分别负责规划、实现、自我调试、测试和硬件性能分析。研究人员在 TPU 基准套件 JaxBench 的 50 项任务，以及先进开源模型的真实工作负载上进行了评估，并称该系统在许多情况下达到了专家手工调优实现的性能。该系统已开源。
本文来源：Hugging Face，仅供学习参考，版权归原作者所有。