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. I received the Meta PhD Fellowship in 2022.

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