MaxKernel: Agentic Kernel Generation for TPUs
arxiv.org原文 ↗
MaxKernel 用规划、实现、自调试、测试和 profiling 子代理串起三种 TPU kernel 搜索:人机协作、自动指标循环和图式全局探索。它在 50 项 JaxBench 任务及真实开源模型 workload 上达到专家手调实现的性能水平;论文的实际贡献在于把编译器反馈变成多代理共同的搜索信号,而不只是让模型一次性写 kernel。
–浏览
评论 · Comments