Haolei Bai

haolei.jpg

👋 Hi, there. I am Haolei Bai (Chinese name: 白浩磊), currently a visiting student at ENCODE Lab, Westlake University, advised by Prof. Huan Wang. Prior to that, I got my M.S. in Signal Processing at Nanyang Technological University, fortune to be advised by Prof. Alex Kot.

I’m interested in Efficient AI (LLMs/VLMs) and Diffusion Language Models (dLLMs).

News


Feb 28, 2025 I graduated from Nanyang Technological University! 🎉
Jan 06, 2025 I joined ENCODE Lab at Westlake University as a visiting student! 🔥

Publications & Preprints

* equal contribution     † corresponding author


  1. arXiv’26
    mobilekernel.png
    MobileKernelBench: Can LLMs Write Efficient Kernels for Mobile Devices?
    Xingze Zou*, Jing Wang*, Yuhua Zheng, and 8 more authors
    arXiv preprint arXiv:2602.11715, 2026
  2. arXiv’26
    dice.png
    DICE: Diffusion Large Language Models Excel at Generating CUDA Kernels
    Haolei Bai, Lingcheng Kong, Xueyi Chen, and 3 more authors
    arXiv preprint arXiv:2602.11715, 2026
  3. CPAL’26
    erc-svd.png
    ERC-SVD: Error-Controlled SVD for Large Language Model Compression
    Haolei Bai, Siyong Jian, Tuo Liang, and 2 more authors
    In Third Conference on Parsimony and Learning, 2026