About

I am a Ph.D. candidate in Computer Science at the University of Science and Technology of China, advised by Prof. Yanyong Zhang and working closely with Prof. Yu Zhang. My research interests include machine learning systems and deep learning compilers. I received my B.E. degree from Chongqing University of Science and Technology in 2016.

Publications

Enabling Tensor Language Model to Assist in Generating High-Performance Tensor Programs for Deep Learning

  • Yi Zhai, Sijia Yang, Keyu Pan, Renwei Zhang, Shuo Liu, Chao Liu, Zichun Ye, Jianmin Ji, Jie Zhao, Yu Zhang, and Yanyong Zhang
  • OSDI 2024

  • | Paper | Code | Slides | Talk |

TLP: A Deep Learning-Based Cost Model for Tensor Program Tuning

  • Yi Zhai, Yu Zhang, Shuo Liu, Xiaomeng Chu, Jie Peng, Jianmin Ji, and Yanyong Zhang
  • ASPLOS 2023

  • | Paper | Code | Slides | Talk |

Crop: An Analytical Cost Model for Cross-Platform Performance Prediction of Tensor Programs

  • Xinyu Sun, Yu Zhang, Shuo Liu, Yi Zhai
  • DAC 2024

Information Fusion Channel Pruning for Accelerating Neural Networks

  • Zhangbin Zhu, Yi Zhai, Yanyong Zhang
  • BigCom 2023


Work & Internships

  • 2023.2 - 2023.12      Research Intern      Huawei Technologies Co., Ltd.
  • 2016.7 - 2018.12      iOS Software Engineer     

Teaching

  • Spring 2021      TA      Computer Architecture, USTC, Undergraduate Course
  • Spring 2020      TA      Computer Architecture, USTC, Undergraduate Course