TLP: A Deep Learning-Based Cost Model for Tensor Program Tuning. - Yi Zhai 005; Yu Zhang 086; Shuo Liu 019; Xiaomeng Chu; Jie Peng 002; Jianmin Ji; Yanyong Zhang (ASPLOS (2), 2023)

Information Fusion Channel Pruning for Accelerating Neural Networks. - Zhangbin Zhu; Yi Zhai 005; Yanyong Zhang (BIGCOM, 2023)

Crop: An Analytical Cost Model for Cross-Platform Performance Prediction of Tensor Programs. - Xinyu Sun; Yu Zhang 086; Shuo Liu 019; Yi Zhai 005 (DAC, 2024)

Enabling Tensor Language Model to Assist in Generating High-Performance Tensor Programs for Deep Learning. - Yi Zhai 005; Sijia Yang; Keyu Pan; Renwei Zhang; Shuo Liu 019; Chao Liu; Zichun Ye; Jianmin Ji; Jie Zhao 002; Yu Zhang 086; Yanyong Zhang (OSDI, 2024)

(X )-evolve: Solution space evolution powered by large language models. - Yi Zhai 005; Zhiqiang Wei; Ruohan Li; Keyu Pan; Shuo Liu 019; Lu Zhang; Jianmin Ji; Wuyang Zhang; Yu Zhang 086; Yanyong Zhang (CoRR, 2025)