MaxKernel Open-Sources a Multi-Agent TPU Kernel Generator Evaluated on 50 JaxBench Tasks

arXiv 2609.04523·medium signalXBlueskyLinkedInCopy link

arXiv 2609.04523 presents MaxKernel, a multi-agent system implementing three paradigms for TPU kernel development: a human-in-the-loop agent for step-by-step design, an autonomous agent running a metric- and trace-driven optimization loop, and a graph-based autonomous search that scales the autonomous agent for global design-space exploration. All three share a pool of specialized sub-agents handling planning, implementation, self-debugging, testing, and hardware profiling. Evaluated on JaxBench (50 TPU kernel tasks) plus real workloads from open-source models, it matches expert hand-tuned baselines; the agent is open-sourced.