Automated High-Performance GPU Kernel Generation
K-Search is an automated kernel engineering system that uses large language models (GPT-5, Gemini etc.) to iteratively generate and optimize GPU kernels. Unlike one-shot code generation, K-Search maintains a co-evolving world model — a structured search tree that encodes hypotheses about kernel bottlenecks, design alternatives, and optimization strategies — guiding multi-round, evidence-driven search over the kernel design space efficiently.
Contributors
Shiyi Cao, Ziming Mao, Joseph E. Gonzalez, Ion Stoica
Publications
CoRR – K-Search: LLM Kernel Generation via Co-Evolving Intrinsic World Model