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 was a visiting student researcher in Sky Lab at UC Berkeley, and 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]

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
  • 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