PIMCOMP: A Universal Compilation Framework for Crossbar-based PIM DNN Accelerators. - Xiaotian Sun; Xinyu Wang; Wanqian Li; Lei Wang 222; Yinhe Han 001; Xiaoming Chen 003 (DAC, 2023)

Efficient GPU Kernels for N: M-Sparse Weights in Deep Learning. - Bin Lin; Ningxin Zheng; Lei Wang 222; Shijie Cao; Lingxiao Ma; Quanlu Zhang; Yi Zhu; Ting Cao; Jilong Xue; Yuqing Yang 001; Fan Yang 024 (MLSys, 2023)

PrimePar: Efficient Spatial-temporal Tensor Partitioning for Large Transformer Model Training. - Haoran Wang 012; Lei Wang 222; Haobo Xu; Ying Wang 001; Yuming Li; Yinhe Han 001 (ASPLOS (3), 2024)

PIMSYN: Synthesizing Processing-in-Memory CNN Accelerators. - Wanqian Li; Xiaotian Sun; Xinyu Wang; Lei Wang 222; Yinhe Han 001; Xiaoming Chen 003 (DATE, 2024)

Ladder: Enabling Efficient Low-Precision Deep Learning Computing through Hardware-aware Tensor Transformation. - Lei Wang 222; Lingxiao Ma; Shijie Cao; Quanlu Zhang; Jilong Xue; Yining Shi 001; Ningxin Zheng; Ziming Miao; Fan Yang 024; Ting Cao; Yuqing Yang 001; Mao Yang 004 (OSDI, 2024)

ConvStencil: Transform Stencil Computation to Matrix Multiplication on Tensor Cores. - Yuetao Chen; Kun Li 016; Yuhao Wang; Donglin Bai; Lei Wang 222; Lingxiao Ma; Liang Yuan; Yunquan Zhang; Ting Cao; Mao Yang 004 (PPoPP, 2024)

The Era of 1-bit LLMs: All Large Language Models are in 1.58 Bits. - Shuming Ma; Hongyu Wang 009; Lingxiao Ma; Lei Wang 222; Wenhui Wang 003; Shaohan Huang; Li Dong 004; Ruiping Wang; Jilong Xue; Furu Wei (CoRR, 2024)

LUT Tensor Core: Lookup Table Enables Efficient Low-Bit LLM Inference Acceleration. - Zhiwen Mo; Lei Wang 222; Jianyu Wei; Zhichen Zeng 002; Shijie Cao; Lingxiao Ma; Naifeng Jing; Ting Cao; Jilong Xue; Fan Yang 024; Mao Yang 004 (CoRR, 2024)

T-MAC: CPU Renaissance via Table Lookup for Low-Bit LLM Deployment on Edge. - Jianyu Wei; Shijie Cao; Ting Cao; Lingxiao Ma; Lei Wang 222; Yanyong Zhang; Mao Yang 004 (EuroSys, 2025)

LUT Tensor Core: A Software-Hardware Co-Design for LUT-Based Low-Bit LLM Inference. - Zhiwen Mo; Lei Wang 222; Jianyu Wei; Zhichen Zeng 002; Shijie Cao; Lingxiao Ma; Naifeng Jing; Ting Cao; Jilong Xue; Fan Yang 024; Mao Yang 004 (ISCA, 2025)

PipeThreader: Software-Defined Pipelining for Efficient DNN Execution. - Yu Cheng; Lei Wang 222; Yining Shi 001; Yuqing Xia; Lingxiao Ma; Jilong Xue; Yang Wang 053; Zhiwen Mo; Feiyang Chen; Fan Yang 024; Mao Yang 004; Zhi Yang 001 (OSDI, 2025)

AttentionEngine: A Versatile Framework for Efficient Attention Mechanisms on Diverse Hardware Platforms. - Feiyang Chen; Yu Cheng; Lei Wang 222; Yuqing Xia; Ziming Miao; Lingxiao Ma; Fan Yang 024; Jilong Xue; Zhi Yang 001; Mao Yang 004; Haibo Chen (CoRR, 2025)

TileLang: A Composable Tiled Programming Model for AI Systems. - Lei Wang 222; Yu Cheng; Yining Shi 001; Zhengju Tang; Zhiwen Mo; Wenhao Xie; Lingxiao Ma; Yuqing Xia; Jilong Xue; Fan Yang 024; Zhi Yang 001 (CoRR, 2025)

SeerAttention-R: Sparse Attention Adaptation for Long Reasoning. - Yizhao Gao 002; Shuming Guo; Shijie Cao; Yuqing Xia; Yu Cheng; Lei Wang 222; Lingxiao Ma; Yutao Sun; Tianzhu Ye; Li Dong 004; Hayden Kwok-Hay So; Yu Hua; et al. (CoRR, 2025)