About

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 machine learning systems, compilers, and distributed systems. 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 am fortunate to be awarded the Meta PhD Fellowship.

Research

• Systems for training and serving large models
  • Projects: Alpa, FastChat, FlexGen
  • Publications: OSDI 22, OSDI 23, ICML 23, MLSys 23.
• Tensor program optimizations
  • Projects: TVM
  • Publications: OSDI 20, NeurIPS 21, OSDI 18, NeurIPS 20, ASPLOS 23.

More publications