Zimu (Zim) Gong

M.S. CSE @ UM-Ann Arbor | ex Research Intern @ Waabi | Lightwheel's first intern

I am a second-year Computer Science and Engineering master's student at the University of Michigan-Ann Arbor. I earned my B.S. in Electrical and Computer Engineering at Shanghai Jiao Tong University.

I was a research intern at Waabi with the A&A behaviors team, working on traffic simulation with flow matching and VAE methods. I was also the first intern at Lightwheel, a start-up company committed to creating sustainable and high-quality 3D data fuel for AI to enter the physical world, where I worked on end-to-end autonomous driving, simulation, imitation learning and reinforcement learning of robot manipulation.

I am interested in VLA models, flow matching, reinforcement learning, and simulation in autonomous driving and robotics. I have hands-on experience in building data pipelines, designing and training machine learning models, conducting closed-loop evaluations, and exercising sim2real domain transfer.