I am currently a machine learning software engineer at Google Cloud LLM DevX team led by Dr. Alex Shraer. Starting 2025, I am one of the main contributors of MaxText, an open-sourced LLM training framework in JAX. Our goal is to develop high-performance, scalable, and easy-to-use codebase for the community.

Before this job, I received my PhD in applied mathematics from the University of Colorado, Boulder, advised by Dr. Alireza Doostan and Dr. Stephen Becker. My research focused on uncertainty quantification and numerical optimization.

News

• 2026 February: Our work Fast algorithms for monotone lower subsets of Kronecker least squares problems is accepted by Journal of Scientific Computing.

• 2025 September: Our work Stochastic Subspace Descent Accelerated via Bi-fidelity Line Search is accepted by TMLR.

• 2025 September: Our work Langevin bi-fidelity importance sampling for failure probability estimation is accepted by Journal of Mechanical Design.

• 2025 May: Our work Exploring Exploration in Bayesian Optimization is accepted by UAI 2025.

• 2025 May: Our work A Unified Framework for Entropy Search and Expected Improvement in Bayesian Optimization is selected as Oral (~1.0%) at ICML 2025.