FlexTensor: An Automatic Schedule Exploration and Optimization Framework for Tensor Computation on Heterogeneous System. - Size Zheng 001; Yun Liang 001; Shuo Wang 009; Renze Chen; Kaiwen Sheng (ASPLOS, 2020)

AMOS: enabling automatic mapping for tensor computations on spatial accelerators with hardware abstraction. - Size Zheng 001; Renze Chen; Anjiang Wei; Yicheng Jin; Qin Han; Liqiang Lu; Bingyang Wu; Xiuhong Li; Shengen Yan; Yun Liang 001 (ISCA, 2022)

NeoFlow: A Flexible Framework for Enabling Efficient Compilation for High Performance DNN Training. - Size Zheng 001; Renze Chen; Yicheng Jin; Anjiang Wei; Bingyang Wu; Xiuhong Li; Shengen Yan; Yun Liang 001 (IEEE Trans. Parallel Distributed Syst., 2022)

Chimera: An Analytical Optimizing Framework for Effective Compute-intensive Operators Fusion. - Size Zheng 001; Siyuan Chen 007; Peidi Song; Renze Chen; Xiuhong Li; Shengen Yan; Dahua Lin; Jingwen Leng; Yun Liang 001 (HPCA, 2023)

MAGIS: Memory Optimization via Coordinated Graph Transformation and Scheduling for DNN. - Renze Chen; Zijian Ding; Size Zheng 001; Chengrui Zhang; Jingwen Leng; Xuanzhe Liu; Yun Liang 001 (ASPLOS (3), 2024)

MoteNN: Memory Optimization via Fine-grained Scheduling for Deep Neural Networks on Tiny Devices. - Renze Chen; Zijian Ding; Size Zheng 001; Meng Li 004; Yun Liang 001 (DAC, 2024)

MCUBERT: Memory-Efficient BERT Inference on Commodity Microcontrollers. - Zebin Yang; Renze Chen; Taiqiang Wu; Ngai Wong 001; Yun Liang 001; Runsheng Wang; Ru Huang 001; Meng Li 004 (ICCAD, 2024)

FlexHE: A flexible Kernel Generation Framework for Homomorphic Encryption-Based Private Inference. - Jiangrui Yu; Wenxuan Zeng; Tianshi Xu; Renze Chen; Yun Liang 001; Runsheng Wang; Ru Huang 001; Meng Li 004 (ICCAD, 2024)

vMCU: Coordinated Memory Management and Kernel Optimization for DNN Inference on MCUs. - Size Zheng 001; Renze Chen; Meng Li 004; Zihao Ye 001; Luis Ceze; Yun Liang 001 (MLSys, 2024)

ArkVale: Efficient Generative LLM Inference with Recallable Key-Value Eviction. - Renze Chen; Zhuofeng Wang; Beiquan Cao; Tong Wu; Size Zheng 001; Xiuhong Li; Xuechao Wei; Shengen Yan; Meng Li 004; Yun Liang 001 (NeurIPS, 2024)

Triton-distributed: Programming Overlapping Kernels on Distributed AI Systems with the Triton Compiler. - Size Zheng 001; Wenlei Bao; Qi Hou; Xuegui Zheng; Jin Fang; Chenhui Huang; Tianqi Li; Haojie Duanmu; Renze Chen; Ruifan Xu; Yifan Guo; Ningxin Zheng; et al. (CoRR, 2025)