Lei Wang

19 papers A* 4A 2B 2Journal 9Unranked 2
YearRankTypeTitle / Venue / Authors
2026 B conf
MetaAttention: A Unified and Performant Attention Framework across Hardware Backends.
PPoPP
Feiyang Chen, Yu Cheng, Lei Wang, Yuqing Xia, Ziming Miao, Lingxiao Ma, Fan Yang, Jilong Xue, Zhi Yang, Mao Yang, Xingda Wei, Haibo Chen
2025 J jnl
AttentionEngine: A Versatile Framework for Efficient Attention Mechanisms on Diverse Hardware Platforms.
CoRR
Feiyang Chen, Yu Cheng, Lei Wang, Yuqing Xia, Ziming Miao, Lingxiao Ma, Fan Yang, Jilong Xue, Zhi Yang, Mao Yang, Haibo Chen
2025 J jnl
BitNet: 1-bit Pre-training for Large Language Models.
J. Mach. Learn. Res.
Hongyu Wang, Shuming Ma, Lingxiao Ma, Lei Wang, Wenhui Wang, Li Dong, Shaohan Huang, Huaijie Wang, Jilong Xue, Ruiping Wang, Yi Wu, Furu Wei
2025 A* conf
LUT Tensor Core: A Software-Hardware Co-Design for LUT-Based Low-Bit LLM Inference.
ISCA
Zhiwen Mo, Lei Wang, Jianyu Wei, Zhichen Zeng, Shijie Cao, Lingxiao Ma, Naifeng Jing, Ting Cao, Jilong Xue, Fan Yang, Mao Yang
2025 A* conf
PipeThreader: Software-Defined Pipelining for Efficient DNN Execution.
OSDI
Yu Cheng, Lei Wang, Yining Shi, Yuqing Xia, Lingxiao Ma, Jilong Xue, Yang Wang, Zhiwen Mo, Feiyang Chen, Fan Yang, Mao Yang, Zhi Yang
2025 J jnl
SeerAttention-R: Sparse Attention Adaptation for Long Reasoning.
CoRR
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
2025 A conf
T-MAC: CPU Renaissance via Table Lookup for Low-Bit LLM Deployment on Edge.
EuroSys
Jianyu Wei, Shijie Cao, Ting Cao, Lingxiao Ma, Lei Wang, Yanyong Zhang, Mao Yang
2025 J jnl
TileLang: A Composable Tiled Programming Model for AI Systems.
CoRR
Lei Wang, Yu Cheng, Yining Shi, Zhengju Tang, Zhiwen Mo, Wenhao Xie, Lingxiao Ma, Yuqing Xia, Jilong Xue, Fan Yang, Zhi Yang
2024 B conf
ConvStencil: Transform Stencil Computation to Matrix Multiplication on Tensor Cores.
PPoPP
Yuetao Chen, Kun Li, Yuhao Wang, Donglin Bai, Lei Wang, Lingxiao Ma, Liang Yuan, Yunquan Zhang, Ting Cao, Mao Yang
2024 J jnl
LUT Tensor Core: Lookup Table Enables Efficient Low-Bit LLM Inference Acceleration.
CoRR
Zhiwen Mo, Lei Wang, Jianyu Wei, Zhichen Zeng, Shijie Cao, Lingxiao Ma, Naifeng Jing, Ting Cao, Jilong Xue, Fan Yang, Mao Yang
2024 A* conf
Ladder: Enabling Efficient Low-Precision Deep Learning Computing through Hardware-aware Tensor Transformation.
OSDI
Lei Wang, Lingxiao Ma, Shijie Cao, Quanlu Zhang, Jilong Xue, Yining Shi, Ningxin Zheng, Ziming Miao, Fan Yang, Ting Cao, Yuqing Yang, Mao Yang
2024 A conf
PIMSYN: Synthesizing Processing-in-Memory CNN Accelerators.
DATE
Wanqian Li, Xiaotian Sun, Xinyu Wang, Lei Wang, Yinhe Han, Xiaoming Chen
2024 J jnl
PIMSYN: Synthesizing Processing-in-memory CNN Accelerators.
CoRR
Wanqian Li, Xiaotian Sun, Xinyu Wang, Lei Wang, Yinhe Han, Xiaoming Chen
2024 — conf
PrimePar: Efficient Spatial-temporal Tensor Partitioning for Large Transformer Model Training.
ASPLOS (3)
Haoran Wang, Lei Wang, Haobo Xu, Ying Wang, Yuming Li, Yinhe Han
2024 J jnl
T-MAC: CPU Renaissance via Table Lookup for Low-Bit LLM Deployment on Edge.
CoRR
Jianyu Wei, Shijie Cao, Ting Cao, Lingxiao Ma, Lei Wang, Yanyong Zhang, Mao Yang
2024 J jnl
The Era of 1-bit LLMs: All Large Language Models are in 1.58 Bits.
CoRR
Shuming Ma, Hongyu Wang, Lingxiao Ma, Lei Wang, Wenhui Wang, Shaohan Huang, Li Dong, Ruiping Wang, Jilong Xue, Furu Wei
2023 — conf
Efficient GPU Kernels for N: M-Sparse Weights in Deep Learning.
MLSys
Bin Lin, Ningxin Zheng, Lei Wang, Shijie Cao, Lingxiao Ma, Quanlu Zhang, Yi Zhu, Ting Cao, Jilong Xue, Yuqing Yang, Fan Yang
2023 A* conf
PIMCOMP: A Universal Compilation Framework for Crossbar-based PIM DNN Accelerators.
DAC
Xiaotian Sun, Xinyu Wang, Wanqian Li, Lei Wang, Yinhe Han, Xiaoming Chen
2023 J jnl
PIMCOMP: A Universal Compilation Framework for Crossbar-based PIM DNN Accelerators.
CoRR
Xiaotian Sun, Xinyu Wang, Wanqian Li, Lei Wang, Yinhe Han, Xiaoming Chen