Amazon Web Services (AWS), Annapurna Labs (2026 to present)
Sr. HPC/AI Engineer. I work on GPU communication performance on large GPU clusters and optimize kernels for mixture-of-experts expert parallelism, and with NVIDIA's NCCL team, I co-lead the work to run NCCL EP on EFA. I co-led DeepEP v2 on EFA and led the NVSHMEM work for DeepEP v1 with NVIDIA's NVSHMEM team, where the changes shipped in NVSHMEM 3.7.0.
Experience
Google (2024 to 2026)
Software Engineer. I led AI-assisted workflows for large-scale code changes and AI-powered integration test setups. I built multi-agent frameworks for agentic code generation and launched alert-based investigations in Google Cloud Assist.
Software Engineer. I led AI-assisted workflows for large-scale code changes and AI-powered integration test setups. I built multi-agent frameworks for agentic code generation and launched alert-based investigations in Google Cloud Assist.
MIT (2023 to present)
Research Affiliate. With Jinhua Zhao and Haris Koutsopoulos, I work on reinforcement learning to optimize complex systems using HPC-based simulators and on LLM agents for cloud anomaly detection. Since 2026, I have also built self-evolving agents with the reef team and co-authored CORAL (COLM 2026).
Research Affiliate. With Jinhua Zhao and Haris Koutsopoulos, I work on reinforcement learning to optimize complex systems using HPC-based simulators and on LLM agents for cloud anomaly detection. Since 2026, I have also built self-evolving agents with the reef team and co-authored CORAL (COLM 2026).
UC Berkeley (2022 to 2025)
Graduate Student Researcher at the Aviation Innovation Research (AIR) Lab and the Cal Unmanned Lab (2022 to 2024). Researched HPC with James Demmel and Raja Sengupta using GPU-based multimodal time-driven simulation (2022 to 2025).
Graduate Student Researcher at the Aviation Innovation Research (AIR) Lab and the Cal Unmanned Lab (2022 to 2024). Researched HPC with James Demmel and Raja Sengupta using GPU-based multimodal time-driven simulation (2022 to 2025).
University of Washington STAR Lab (2022 to 2023)
Research Assistant. Worked on TinT for city-wide congestion prediction.
Research Assistant. Worked on TinT for city-wide congestion prediction.
Lawrence Berkeley National Laboratory (2021 to 2023)
Graduate Student Researcher.
Graduate Student Researcher.
Education
UC Berkeley (2020 to 2024)
Ph.D. (2021 to 2024) and M.S. (2020 to 2021), advised by James Demmel (SLICE Lab) and Raja Sengupta (Cal Unmanned Lab). Alexandre Bayen (BAIR) served on my committee.
Ph.D. (2021 to 2024) and M.S. (2020 to 2021), advised by James Demmel (SLICE Lab) and Raja Sengupta (Cal Unmanned Lab). Alexandre Bayen (BAIR) served on my committee.
Tongji University (2016 to 2020)
B.S.
B.S.
Honors
- [2024] ICRAT Best Paper Award, Trajectories and Networks track.
- [2023] ASCE ICTD AI in Transportation Committee Outstanding Session Organizer Award.
- [2022] NSF AI Workshop Phase II Travel Award.
- [2021] Joseph M. Sussman Best Paper Prize.
- [2020] Gold Award, 6th China International "Internet+" College Students' Innovation and Entrepreneurship Competition.
- [2019] Outstanding Student of Shanghai.
- [2018] Shanghai Municipal Innovation and Entrepreneurship Project Award.
- [2017] Outstanding Student Leader, Tongji University.
Teaching
UC Berkeley (Fall 2022)
Graduate Student Instructor for Control and Information Management.
Graduate Student Instructor for Control and Information Management.
UC Berkeley (2023 to 2024)
Teaching Assistant for Emerging Technologies for Public Health.
Teaching Assistant for Emerging Technologies for Public Health.
Service
Frontiers in Built Environment (2024)
Topic Editor for the research topic "Advanced Technologies for Aviation Operations and Passenger Experience".
Topic Editor for the research topic "Advanced Technologies for Aviation Operations and Passenger Experience".
ASCE (2023 to 2024)
Member of the Artificial Intelligence in Transportation Committee and vice chair of its Student Subcommittee.
Member of the Artificial Intelligence in Transportation Committee and vice chair of its Student Subcommittee.
ASCE (2023 to 2024)
Member of the Active Transportation Committee.
Member of the Active Transportation Committee.
Journal reviewing: IEEE Transactions on Intelligent Transportation Systems, IEEE Transactions on Industrial Informatics, Scientific Reports, Frontiers in Psychology.
Conference reviewing: ICLR 2026, ICML 2026, CVPR 2026, ECCV 2026, COLM 2026, NeurIPS 2025, AAAI 2025, ITSC 2025, TRB Annual Meeting 2025.
Earlier Work
Before moving to the US, I helped Zihao (同济子豪兄) build his AI popular-science channel on Bilibili. The channel now has more than 350,000 followers (October 2026). Bilibili channel