Summary
Jiahao CUI is a software engineer and Georgia Tech master's student focusing on distributed systems and compiler optimizations for machine learning. With nine years of industry and research experience across cloud, VR, and HPC domains, he currently develops JAX Mixture-of-Experts models on TPUs at Google. His research at Georgia Tech has produced PyTorch compiler optimizations and ML-driven algorithm selection for SMT solvers, reflecting a blend of systems-level thinking and ML tooling. He has taught advanced operating systems and UX design, demonstrating an ability to communicate complex technical concepts to diverse audiences. Comfortable moving between research, teaching, and production engineering, Jiahao combines rigorous academic training from multiple international universities with hands-on experience shipping high-performance ML systems.
9 years of coding experience
1 year of employment as a software developer
Charles III University of Madrid (Universidad Carlos III de Madrid)
Master of Science in Computational Science and Engineering, Master of Science in Computational Science and Engineering at Georgia Institute of Technology
highschool, science, highschool, science at IES velazquez
Computer Science, Exchange, Computer Science, Exchange at University of Waterloo
Spanish, English, Chinese