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)

SuSy: A Programming Model for Productive Construction of High-Performance Systolic Arrays on FPGAs. - Yi-Hsiang Lai; Hongbo Rong; Size Zheng 001; Weihao Zhang; Xiuping Cui; Yunshan Jia; Jie Wang 022; Brendan Sullivan; Zhiru Zhang; Yun Liang 001; Youhui Zhang; Jason Cong; et al. (ICCAD, 2020)

HASCO: Towards Agile HArdware and Software CO-design for Tensor Computation. - Qingcheng Xiao; Size Zheng 001; Bingzhe Wu; Pengcheng Xu 005; Xuehai Qian; Yun Liang 001 (ISCA, 2021)

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)

Rubick: A Synthesis Framework for Spatial Architectures via Dataflow Decomposition. - Zizhang Luo; Liqiang Lu; Size Zheng 001; Jieming Yin; Jason Cong; Jianwei Yin; Yun Liang 001 (DAC, 2023)

Memory and Computation Coordinated Mapping of DNNs onto Complex Heterogeneous SoC. - Size Zheng 001; Siyuan Chen 007; Yun Liang 001 (DAC, 2023)

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)

ARES: A Mapping Framework of DNNs Towards Diverse PIMs with General Abstractions. - Xiuping Cui; Size Zheng 001; Tianyu Jia; Le Ye; Yun Liang 001 (ICCAD, 2023)

TileFlow: A Framework for Modeling Fusion Dataflow via Tree-based Analysis. - Size Zheng 001; Siyuan Chen 007; Siyuan Gao; Liancheng Jia; Guangyu Sun 003; Runsheng Wang; Yun Liang 001 (MICRO, 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)

SpecPIM: Accelerating Speculative Inference on PIM-Enabled System via Architecture-Dataflow Co-Exploration. - Cong Li; Zhe Zhou 002; Size Zheng 001; Jiaxi Zhang 001; Yun Liang 001; Guangyu Sun 003 (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)

SpREM: Exploiting Hamming Sparsity for Fast Quantum Readout Error Mitigation. - Hanyu Zhang; Liqiang Lu; Siwei Tan; Size Zheng 001; Jia Yu; Jianwei Yin (DAC, 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)

Atom: Low-Bit Quantization for Efficient and Accurate LLM Serving. - Yilong Zhao; Chien-Yu Lin; Kan Zhu; Zihao Ye 001; Lequn Chen 001; Size Zheng 001; Luis Ceze; Arvind Krishnamurthy; Tianqi Chen 001; Baris Kasikci (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)

ShadowKV: KV Cache in Shadows for High-Throughput Long-Context LLM Inference. - Hanshi Sun; Li-Wen Chang; Wenlei Bao; Size Zheng 001; Ningxin Zheng; Xin Liu 086; Harry Dong; Yuejie Chi; Beidi Chen (CoRR, 2024)

Rubick: A Unified Infrastructure for Analyzing, Exploring, and Implementing Spatial Architectures via Dataflow Decomposition. - Liqiang Lu; Zizhang Luo; Size Zheng 001; Jieming Yin; Jason Cong; Yun Liang 001; Jianwei Yin (IEEE Trans. Comput. Aided Des. Integr. Circuits Syst., 2024)

Qtenon: Towards Low-Latency Architecture Integration for Accelerating Hybrid Quantum-Classical Computing. - Chenning Tao; Liqiang Lu; Size Zheng 001; Li-Wen Chang; Minghua Shen; Hanyu Zhang; Fangxin Liu; Kaiwen Zhou; Jianwei Yin (ISCA, 2025)

Comet: Fine-grained Computation-communication Overlapping for Mixture-of-Experts. - Shulai Zhang; Ningxin Zheng; Haibin Lin; Ziheng Jiang; Wenlei Bao; Chengquan Jiang; Qi Hou; Weihao Cui; Size Zheng 001; Li-Wen Chang; Quan Chen 002; Xin Liu 086 (CoRR, 2025)

TileLink: Generating Efficient Compute-Communication Overlapping Kernels using Tile-Centric Primitives. - Size Zheng 001; Jin Fang; Xuegui Zheng; Qi Hou; Wenlei Bao; Ningxin Zheng; Ziheng Jiang; Dongyang Wang; Jianxi Ye; Haibin Lin; Li-Wen Chang; Xin Liu 086 (CoRR, 2025)

MoQa: Rethinking MoE Quantization with Multi-stage Data-model Distribution Awareness. - Zihao Zheng; Xiuping Cui; Size Zheng 001; Maoliang Li; Jiayu Chen; Yun (Eric) Liang; Xiang Chen (CoRR, 2025)

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)

MxMoE: Mixed-precision Quantization for MoE with Accuracy and Performance Co-Design. - Haojie Duanmu; Xiuhong Li; Zhihang Yuan; Size Zheng 001; Jiangfei Duan; Xingcheng Zhang; Dahua Lin (CoRR, 2025)

MegaScale-MoE: Large-Scale Communication-Efficient Training of Mixture-of-Experts Models in Production. - Chao Jin; Ziheng Jiang; Zhihao Bai; Zheng Zhong; Juncai Liu; Xiang Li 067; Ningxin Zheng; Xi Wang; Cong Xie; Qi Huang; Wen Heng; Yiyuan Ma; et al. (CoRR, 2025)

SwiftSpec: Ultra-Low Latency LLM Decoding by Scaling Asynchronous Speculative Decoding. - Ziyi Zhang; Ziheng Jiang; Chengquan Jiang; Menghan Yu; Size Zheng 001; Haibin Lin; Henry Hoffmann; Xin Liu 086 (CoRR, 2025)