MaxKernel introduces a multi-agent framework to automate the creation of high-performance custom kernels for TPUs, addressing the complexity of hardware-level optimization. It supports three paradigms: a Human-in-the-Loop agent for collaborative design, an Autonomous agent for metric-driven optimization, and a Graph-Based Agent for broad design space exploration. The system leverages real-time compiler feedback and a shared pool of specialized sub-agents to improve efficiency and performance.
- Automates TPU kernel development using real-time compiler feedback loops.
- Supports Human-in-the-Loop collaboration for step-by-step design refinement.
- Enables fully autonomous optimization driven by performance metrics and traces.
- Scales exploration via Graph-Based Autonomous Search across the design space.