PhD, Carnegie Mellon University

About Me

I am Mengdi Wu (吴梦迪), a fourth-year Ph.D. student in the Computer Science Department at Carnegie Mellon University, advised by Zhihao Jia. I received my undergraduate degree from the Yao Class at Tsinghua University in 2023. I am honored to be supported by the Amazon Fellowship for my research.

I work on machine learning systems, with a focus on ML compilers, superoptimization, and kernel verification. I am interested in understanding the computational structure of machine learning applications and developing techniques to make them more efficient and reliable.

Publications

Identity Testing for Circuits with Exponentiation Gates
Jiatu Li^*, Mengdi Wu^*
ITCS 2026

Collie: Token-Level Co-Serving of LLM Inference and Fine-Tuning with SLO Guarantees
Gabriele Oliaro, Xupeng Miao, Xinhao Cheng, Vineeth Kada, Ruohan Gao, Yingyi Huang, Remi Delacourt, April Yang, Yingcheng Wang, Mengdi Wu, Colin Unger, Zhihao Jia
NSDI 2026

Mirage: A Multi-Level Superoptimizer for Tensor Programs
Mengdi Wu, Xinhao Cheng, Shengyu Liu, Chunan Shi, Jianan Ji, Kit Man Ao, Praveen Velliengiri, Xupeng Miao, Oded Padon, Zhihao Jia
OSDI 2025

GraphPipe: Improving the Performance and Scalability of DNN Training with Graph Pipeline Parallelism
Byungsoo Jeon^*, Mengdi Wu^*, Shiyi Cao^*, Sunghyun Kim^*, Sunghyun Park, Neeraj Aggarwal, Colin Unger, Daiyaan Arfeen, Peiyuan Liao, Xupeng Miao, Mohammad Alizadeh, Gregory R. Ganger, Tianqi Chen, Zhihao Jia
ASPLOS 2025

LACO: A Latency-Constraint Offline Neural Network Scheduler towards Reliable Self-Driving Perception
Zhanhong Tan, Zijian Zhu, Mengdi Wu, Kaisheng Ma
ICCAD 2024

Finding the Task-Optimal Low-Bit Sub-Distribution in Deep Neural Networks
Runpei Dong^*, Zhanhong Tan^*, Mengdi Wu, Linfeng Zhang, Kaisheng Ma
ICML 2022

Awards

National Student Computer System Capability Challenge 2020 (Compiler track), 2nd Prize [Cutepiler]
With Jiatu Li, Runda Liu, and Zhidong Wang.
National Olympiad in Informatics, China, Gold Medal