← Search

Yitao Hu

2 accepted papers

2026

KernelBand: Steering LLM-based Kernel Optimization via Hardware-Aware Multi-Armed Bandits

ICML 2026poster

High-performance GPU kernels are critical for efficient LLM serving, yet their optimization remains a bottleneck requiring deep system expertise. While code LLMs show promise in generating functionally correct code, kernel optimization is intrinsically a search problem over a vast optimization space…

Cited by 0SourceScholar
2026

Mosaic: Unlocking Over 30$\times$ Context Length for Diffusion LLMs Inference via Global Memory Planning and Dynamic Peak Taming

ICML 2026poster

Diffusion-based large language models (dLLMs) have emerged as a promising alternative to autoregressive models, leveraging simultaneous denoising to enable global planning and iterative refinement. These properties make dLLMs particularly attractive for long-context generation. However, deploying dL…

Cited by 0SourceScholar