World Model

GPU Forecasters: Language Models as Selective Surrogates for Kernel Runtime Optimization

LLMs as selective surrogates that forecast GPU kernel performance to accelerate kernel search

Zaid Khan, Justin Chih-Yao Chen, Jaemin Cho, Elias Stengel-Eskin, Mohit Bansal