I am a Ph.D. student in the EECS department at UC Berkeley, advised by Ion Stoica and Joseph E. Gonzalez. My research interests lie in the intersection of machine learning and programming systems, especially domain-specific compilers for accelerated and scalable deep learning.

I received my B.S degree in computer science from ACM Honors Class, Shanghai Jiao Tong University. I was a research intern at Amazon Web Services, OctoML, and University of Washington, working closely with Tianqi Chen, Luis Ceze, and Yida Wang. I am a PMC member of Apache TVM. I received the Meta PhD Fellowship in 2022.

Take a look at Alpa, our recently-released system for automatic model-parallel training of large models.