I am a Ph.D. student in the EECS department at UC Berkeley, advised by Ion Stoica and Joseph E. Gonzalez. My research interests include large-scale machine learning systems, compilers, parallel computing and program synthesis.

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.