MaxKernel: Agentic Kernel Generation for TPUs
This work presents MaxKernel, a multi-agent system that implements three distinct paradigms for TPU kernel development: a Human-in-the-Loop (HITL) agent for collaborative, step-by-step design; an Autonomous (Auto) agent that executes a fully automated, metric/trace-driven optimization loop; and a Graph-Based Autonomous...
Shang-Kun Wang, Nina Cai, Charles Hoong et al.
· 0 citations