MaxKernel launches agentic kernel generation for TPUs
agents
| Source: HF Papers | Original article
Researchers introduce MaxKernel, an agentic system that uses large language models and real-time compiler feedback to automatically generate high‑performance custom TPU kernels.
Google has unveiled **MaxKernel**, an open‑source, multi‑agent system that automatically generates high‑performance TPU kernels. The platform pairs large language models with real‑time compiler feedback, letting an “agentic” search loop iterate on Pallas kernel code until it matches the speed of hand‑tuned expert implementations.
The team evaluated MaxKernel on **JaxBench**, a benchmark suite of 50 diverse TPU kernel tasks, as well as on complex workloads drawn from state‑of‑the‑art open‑source models. Across the board, the generated kernels approached the performance of manually optimized code, demonstrating that the system can replace a traditionally labor‑intensive engineering step with an automated workflow.
Why it matters: Custom kernels are the “secret sauce” behind many recent LLM speedups, yet writing them for TPUs using Pallas has required deep hardware expertise. By automating this process, MaxKernel lowers the barrier for researchers and developers to extract maximum throughput from Google’s accelerator hardware. The open‑source release in the Google Accelerator Agents GitHub repository also invites community contributions, potentially accelerating the broader adoption of TPU‑based AI workloads.
What to watch next: The first public rollout will likely focus on integration with existing JAX‑based pipelines, but the underlying agentic approach could be extended to other accelerator stacks such as GPUs or emerging ASICs. Follow‑up work may explore tighter coupling with tool‑use frameworks we covered earlier, where agents orchestrate end‑to‑end model training and deployment. As we reported on Sep 4 2026, agentic AI is moving beyond chat and reasoning into concrete system‑level tasks; MaxKernel marks a concrete step toward fully automated hardware optimisation, a development that could reshape how AI infrastructure is built and maintained.
Sources
Back to AIPULSEN