Yuhao Ge

I am working at Google, where I focus on improving Gemini training and serving performance on TPU, and enhancing third party TPU experience through TorchTPU.

Previously, I worked as a Software Engineer at Amazon AWS, Annapurna Labs, where I focused on LLM training/inference acceleration on Trainium chips, primarily working in ML Compiler Optimization, Kernel Language Design, and Kernel Optimization.

I was an Master’s student in Computer Science at UIUC, working under Prof. Charith Mendis. My research interests span Machine Learning, Compilers, and LLM Efficiency.

I earned dual B.S. degrees in Computer Engineering from Zhejiang University and the University of Illinois at Urbana-Champaign through their joint program. I also had the opportunity to be a visiting research student at the UCLA VAST Lab, under the supervision of Prof. Jason Cong, where I worked on FPGA accelerator design automation. Additionally, I have interned at TikTok and NFTGo.