Research · Products and Tools
MaxKernel:用于TPU的智能体内核代码生成
MaxKernel是一个多智能体系统,实现了三种TPU内核开发范式,在包含50个多样化任务的JaxBench基准上进行了评估,并已开源。
阅读 Hugging Face Daily Papers 原文为什么重要
该工作展示了多智能体系统在专用硬件内核开发中的潜力,有望加速TPU等加速器的软件生态建设,降低高性能计算开发门槛。
关键事实
事实 1
MaxKernel是一个多智能体系统,实现了三种TPU内核开发范式。
来源与依据
we present MaxKernel, a multi-agent system that implements three distinct paradigms for TPU kernel development
事实 2
MaxKernel在JaxBench上进行了评估,该基准包含50个多样化的TPU内核任务。
来源与依据
We evaluate MaxKernel on JaxBench, a comprehensive suite of 50 diverse kernel tasks for TPUs
事实 3
MaxKernel的智能体是开源的,并可在GitHub上获取。
来源与依据
Our agent is open-sourced and available https://github.com/AI-Hypercomputer/accelerator-agents/tree/main/MaxKernel