Jiawei Zhao

I am a PhD candidate in the CMS department at Caltech, advised by Anima Anandkumar. During my PhD, I spent most of my time as an intern at NVIDIA Research. I am also collaborating with Atlas Wang at UT Austin, Beidi Chen at CMU, Florian Schäfer at Georgia Tech, and Yuandong Tian at Meta AI (FAIR).

I am interested in bridging hardware constraints with the principles of learning in neural networks. I focus on developing hardware-efficient learning algorithms that are efficient and scalable for training large-scale models, such as large language models (LLMs). Check out my research for more details.

Reach me at firstname at caltech dot edu