3 papers
cs.AI2026
KernelGenBench: A Multi-Source and Multi-Chip Benchmark for LLM-based Kernel Generation
Peiyu Zang, Jian Tao, Jialing Zhang +4
Large language models (LLMs) have significantly increased the demand for efficient accelerator kernels, but kernel development remains a highly specialized and labor-intensive task…
cs.LG2026
FlashPDE: A Drop-In Fused Triton Operator Library for Neural PDE Solvers
Peiyu Zang, Bosen Xie, Ruoxiang Xu +1
Physics-Informed Neural Networks (PINNs) solve PDEs by incorporating physical constraints into neural-network training, but large-scale problems are limited by automatic-differenti…
cs.LG2026
Towards Automated Kernel Generation in the Era of LLMs
Yang Yu, Peiyu Zang, Chi Hsu Tsai +11
The performance of modern AI systems is fundamentally constrained by the quality of their underlying GPU kernels, which translate high-level algorithmic semantics into low-level ha…