Lei Wang

PhD student, Peking University

Joined

May 2025

Names

Lei Wang (Preferred)

Emails

****@outlook.com (Confirmed)

,

****@stu.pku.edu.cn (Confirmed)

,

****@stu.pku.edu.cn (Confirmed)

Personal Links

Career & Education History

PhD student

Peking University (stu.pku.edu.cn)

2025 – Present

MS student

Computing Technology, University of the Chinese Academy of Sciences (mails.ucas.ac.cn)

2021 – 2024

Advisors, Relations & Conflicts

No relations added

Expertise

Machine Learning System

2021 – 2025

Publications

Sparse Attention Adaptation for Long Reasoning
Yizhao Gao, Shuming Guo, Shijie Cao, Yuqing Xia, Yu Cheng, Lei Wang, Lingxiao Ma, Yutao Sun, Tianzhu Ye, Li Dong, Hayden Kwok-Hay So, Yu Hua, Ting Cao, Fan Yang, Mao Yang
- ICLR 2026 Poster
- Readers: Everyone
TileLang: Bridge Programmability and Performance in Modern Neural Kernels
Lei Wang, Yu Cheng, Yining Shi, Zhiwen Mo, Zhengju Tang, Wenhao Xie, Tong Wu, Lingxiao Ma, Yuqing Xia, Jilong Xue, Fan Yang, Zhi Yang
- ICLR 2026 Oral
- Readers: Everyone
T-MAC: CPU Renaissance via Table Lookup for Low-Bit LLM Deployment on Edge
Jianyu Wei, Shijie Cao, Ting Cao, Lingxiao Ma, Lei Wang, Yanyong Zhang, Mao Yang
- EuroSys 2025
- Readers: Everyone
Ladder: Enabling Efficient Low-Precision Deep Learning Computing through Hardware-aware Tensor Transformation
Lei Wang, Lingxiao Ma, Shijie Cao, Quanlu Zhang, Jilong Xue, Yining Shi, Ningxin Zheng, Ziming Miao, Fan Yang, Ting Cao, Yuqing Yang, Mao Yang
- OSDI 2024
- Readers: Everyone
PrimePar: Efficient Spatial-temporal Tensor Partitioning for Large Transformer Model Training
Haoran Wang, Lei Wang, Haobo Xu, Ying Wang, Yuming Li, Yinhe Han
- ASPLOS (3) 2024
- Readers: Everyone
LUT Tensor Core: Lookup Table Enables Efficient Low-Bit LLM Inference Acceleration
Zhiwen Mo, Lei Wang, Jianyu Wei, Zhichen Zeng, Shijie Cao, Lingxiao Ma, Naifeng Jing, Ting Cao, Jilong Xue, Fan Yang, Mao Yang
- CoRR 2024
- Readers: Everyone

Co-Authors

View all 34 co-authors