I am a fourth-year Ph.D. student in the Department of Computer Science and Technology, Tsinghua University. I am currently advised by Jidong Zhai in the PACMAN group. I received my B.Eng. degree from Tsinghua University in 2021. I am now visiting the Sky Lab at UC Berkeley, and is working with Prof. Ion Stoica.

Research

I work on deep learning systems, with a particular interest in compilers & runtime systems for irregular deep learning workloads, such as complex Python DNN programs and dynamic control flows. Additionally, I used to work on efficient quantum circuit simulation and general parallel computing.

Selected Publications

  • [ATC’24] MagPy: Compiling Eager Mode DNN Programs by Monitoring Execution States
    Chen Zhang, Rongchao Dong, Haojie Wang, Runxin Zhong, Jike Chen, Jidong Zhai
    [Paper] [Code]
  • [OSDI’23] Cocktailer: Analyzing and Optimizing Dynamic Control Flow in Deep Learning
    Chen Zhang, Lingxiao Ma, Jilong Xue, Yining Shi, Ziming Miao, Fan Yang, Jidong Zhai, Zhi Yang, Mao Yang
    [Paper] [Code]
  • [SC’22] UniQ: A Unified Programming Model for Efficient Quantum Circuit Simulation
    Chen Zhang, Haojie Wang, Zixuan Ma, Lei Xie, Zeyu Song, Jidong Zhai
    [Paper] [Code]
  • [ICS’21] HyQuas: Hybrid Partitioner Based Quantum Circuit Simulation System on GPU
    Chen Zhang, Zeyu Song, Haojie Wang, Kaiyuan Rong, Jidong Zhai
    Win the Best Student Paper award
    [Paper] [Code]
  • [TPDS’21] Critique of “Planetary Normal Mode Computation: Parallel Algorithms, Performance, and Reproducibility” by SCC Team From Tsinghua University
    Chen Zhang, Chenggang Zhao, Jiaao He, Shengqi Chen, Liyan Zheng, Kezhao Huang, Wentao Han, Jidong Zhai
    [Paper] [Code]
  • [SC’24] MIXQ: Taming Dynamic Outliers in Mixed-Precision Quantization by Online Prediction
    Yidong Chen, Chen Zhang, Rongchao Dong, Haoyuan Zhang, Yonghua Zhang, Zhonghua Lu, Jidong Zhai
    Paper will be available soon
  • [OSDI’22] Roller: Fast and efficient tensor compilation for deep learning
    Hongyu Zhu, Ruofan Wu, Yijia Diao, Shanbin Ke, Haoyu Li, Chen Zhang, Jilong Xue, Lingxiao Ma, Yuqing Xia, Wei Cui, Fan Yang, Mao Yang, Lidong Zhou, Asaf Cidon, Gennady Pekhimenko
    [Paper]
  • [PLDI’22] FreeTensor: a free-form DSL with holistic optimizations for irregular tensor programs
    Shizhi Tang, Jidong Zhai, Haojie Wang, Lin Jiang, Liyan Zheng, Zhenhao Yuan, Chen Zhang
    [Paper]
  • [ICS’22] Efficiently emulating high-bitwidth computation with low-bitwidth hardware
    Zixuan Ma, Haojie Wang, Guanyu Feng, Chen Zhang, Lei Xie, Jiaao He, Shengqi Chen, Jidong Zhai
    [Paper]
  • [PPoPP’22] PerFlow: A domain specific framework for automatic performance analysis of parallel applications
    Yuyang Jin, Haojie Wang, Runxin Zhong, Chen Zhang, Jidong Zhai
    [Paper]
  • [TC’21] A Fast Lock for Explicit Message Passing Architectures
    Xiongchao Tang, Chen Zhang, Jidong Zhai, Xuehai Qian, Wenguang Chen, Yong Jiang
    [Paper]

Honors

  • Special-Class Scholarship of Tsinghua University (Top 0.3%) 2020.11
    • The highest honor for undergraduates at Tsinghua, only 10 students per year
  • Beijing Outstanding Undergraduate Thesis (Top 0.5%)* 2021.06
  • National Scholarship (Top 1%) 2019.10 & 2020.10 & 2023.10
  • CCF Outstanding Undergraduate: 98 students in China 2020.10

Competitions

  • ISC21 Student Cluster Competition: Champion & Best Linpack (team captain) 2021.06
  • SC20 Student Cluster Competition: Champion & Best Linpack (team captain) 2020.11
  • SC19 Student Cluster Competition: Champion 2019.11
  • China Collegiate Programming Contest (Girls Only): Champion, solve all problems 2019.05
  • The 2018 ICPC Asia Jiaozuo Regional Contest: Rank 2, the first to solve problem C 2018.11
  • National Olympiad in Informatics: Silver medal 2016.07

Talks

Cocktailer: Analyzing and Optimizing Dynamic Control Flow in Deep Learning

  • OSDI2023, ChinaSys 2023Spring, HPCChina 2023, ETH AST Lab 2024

MagPy: Compiling Eager Mode DNN Programs by Monitoring Execution States

  • ATC2024, ChinaSys 2023Spring

HyQuas: hybrid partitioner based quantum circuit simulation system on GPU

  • ICS2021, HPCChina2022

UniQ: a unified programming model for efficient quantum circuit simulation

  • SC2022, HPCChina2022

Experience on Student Cluster Competitions

  • HPCChina 2021

Teaching

  • TA for Introduction to High Performance Computing (Spring 2021, Spring 2022, Spring 2023, Spring 2024)
  • TA for Introduction to Computer Systems (Fall 2022, Fall 2023)
  • TA for Advanced Compiler (Fall 2021)

Internship

  • Microsoft Research Asia, 2020.07-2021.12 (Mentors: Fan Yang and Lingxiao Ma)
  • Tencent Wechat Group, 2019.07-2019.08