Yiming Zhou

About Me

I’m a Research Scientist at ByteDance Seed, working on compiler-based pre-training infrastructure for large-scale LLMs.

Prior to ByteDance, I was a Research Scientist at Meta on the PyTorch team, where I worked on PyTorch Distributed (GraphTrainer in TorchTitan) and the PyTorch Compiler stack (torch.compile, torch.export, AOTInductor).

Prior to Meta, I earned my Ph.D. in Electrical and Computer Engineering from UCLA under the supervision of Prof. Bahram Jalali. My research there centered on physics-informed machine learning and physics-inspired algorithms, including novel neural architectures and real-time computational imaging systems. Prior to UCLA, I got my B.E. in Electronic Information Engineering from University of Science and Technology of China (USTC).

Powered by Jekyll and Minimal Light theme.