I am a final-year Ph.D. candidate in Electrical Computer Engineering at University of Texas at Austin, under the supervision of Prof. David Z. Pan (Fellow of ACM, IEEE, SPIE) and Prof. Ray T. Chen (Fellow of NAI, IEEE, OSA, SPIE). I work closely with Prof. Zhangyang "Atlas" Wang on efficient AI algorithms at UT Austin and Prof. Jiaqi Gu on emerging ML hardware at ASU.

Previously, I received the B.Eng. degree in Microelectronics Science and Engineering from Shanghai Jiaotong University with highest honor, in 2020. I grew up in a small town in southwest China, where I received all my education till college.

I was recognized as the ML and Systems Rising Star in 2025, received MLSys'25 Outstanding Paper Award (Honorable Mention) and more honors.

Recent Updates:

Our APOLLO received the Outstanding Paper Award (Honorable Mention) at MLSys'25.

I will join Meta as Research Scientist Intern this summer to study efficient large reasoning models.

My research interests lie in Efficient AI computing, focusing on Efficient ML Hardware/Systems and Efficient AI Algorithms with Hardware/Systems Awareness. I firmly believe that the co-design of hardware and algorithms is key to unlocking the ultimate efficiency in AI computing, shaping my expertise across both domains.

• Hardware-Software Co-Design for Emerging AI Hardware and Computing Systems:
  
  • First Photonic Transformer Accelerator [HPCA'24]. Released the first arch-level photonic accelerator simulator
  • Ultra-fast Optical Device Simulation for Photonic Design Automation [NeurIPS'24]
  • Efficient and Robust Photonic AI engine [Journal of Lightwave Technology'24, MLsys'23 SNAP workshop, ICCAD'22, TCAD'22, ASP-DAC'22]
  • Contributor of first photonic AI library TorchONN (PCM-based ONN, on-chip training)
• Efficient Training and Inference for Large Foundation Models:
  
  • Memory-efficient LLM training (APOLLO) with SGD-like memory efficiency but AdamW-level performance [MLSys'25] (🏆 Outstanding Paper Honorable Mention), Efficient On-Chip Training [NeurIPS'22]
  • Efficient Pre-LN Transformers [NeurIPS'23 Spotlight], Memory-efficient Neural Network with in-situ weights generation [ICCV'21]

The University of Texas at Austin Ph.D. in Electrical and Computer Engineering (2020 - 2025) First year (2020-2021) conducted part-time in China due to COVID-19 ML and System Rising star (2025), Graduate School Continuing Fellowship Nomination (1 of 2 in the ECE department), Texas ECE Graduate Achievement Award Advisor: Prof. David Z. Pan(Fellow of ACM, IEEE, SPIE)

Shanghai Jiao Tong University B.Eng. in Microelectronics Science and Engineering (2016 - 2020) Graduated with Highest Honors Overall GPA: 3.81/4.00, Ranking: 2/57

Journal