I am a master's student in the Electrical & Computer Engineering department at Cornell University, where I have the privilege of being advised by Prof. Zhiru Zhang.

My research interests span computer architecture, machine learning systems (MLSys), machine learning, hardware compiler. I am passionate about pushing the boundaries of hardware-software co-design for efficient AI and system architectures.

I am actively seeking a Ph.D. position for Fall 2025 to continue advancing these areas of research.

🔥 News

2024.10: 🎉🎉 Our Paper SmoothE is accepted by ASPLOS 2025!

📝 Publications

ASPLOS 2025

SmoothE: Differentiable E-Graph Extraction

Yaohui Cai, Kaixin Yang, Chenhui Deng, Cunxi Yu, Zhiru Zhang

International Conference on Architectural Support for Programming Languages and Operating Systems (ASPLOS), 2025. [Accept Rate: 12.7%]

[Introduction]

The first differentiable approach to e-graph extraction by relaxing the discrete optimization problem of e-node selection into a continuous, probabilistic formulation which enables parallelizable global optimization that supports any differentiable objectives.
Implement SmoothE in PyTorch and further introduce a set of performance optimization techniques to exploit vectorized processing, sparsity, seed batching to improve the efficiency of GPU execution.

[Code] To be open-sourced soon.

[Paper] Coming up soon!

📖 Educations

Cornell University, Ithaca, New York, United States
Aug 2023 - May 2024
Master of Engineering in Computer Engineering
GPA: 3.96

Beijing Jiaotong University, Beijing, China
Sep 2019 - June 2023
Bachelor of Engineering in Mechatronics
Major GPA: 92.4, Outstanding Thesis Award

💼 Research Experience

Cornell Computer System Lab, Ithaca, New York, United States
May 2024 - Present, Full time, On-site
Aug 2023 - May 2024, Part time, On-site
Position: Research Intern
Advisor: Prof. Zhiru Zhang

💻 Internships

Careverse Technology, Shanghai, China
June 2023 - August 2023
Machine Learning Engineer

NIO, Shanghai, China
February 2023 - June 2023
Embedded Software Engineer

🎖 Honors and Awards

2020-2021 & 2021-2022 Second Class Scholarship for Academic Excellence.
2020-2021 & 2021-2022 Outstanding Student of the Acedamic Year 2020-2021.