I am a Ph.D. candidate in EDIC Computer and Communication Sciences at EPFL, advised by Prof. Sanidhya Kashyap. I received my B.S. in Computer Science with summa cum laude from Turing Class at Peking University. I also minored in Finance.
My research focuses on managing state across the memory hierarchy, such as pages, KV caches, and agent memory, from the OS kernel up to LLM agents. Every layer faces the same trade-off between fetching state and regenerating it, but the right balance differs across layers and shifts with the hardware and workload. I tackle it from two sides:
- Scalable mechanisms that move pages through the hierarchy without becoming a bottleneck.
- Learned policies that decide what to keep, fetch, or regenerate for the workload at hand.
Along the way, I have built systems at every level of the stack, from memory management in the Linux kernel to KV cache management for LLM inference and communication for LLM pretraining. I contribute regularly to open-source projects such as SGLang and Linux (both “kernels”, in a sense). I am also a member of the Swiss Apertus LLM project, where I contribute to pretraining mostly through UCCL.
Education
École Polytechnique Fédérale de Lausanne (EPFL), Switzerland
Sep. 2022 - PresentPh.D. in Computer Science
Advised by Prof. Sanidhya Kashyap
Peking University, China
Sep. 2018 - Jul. 2022B.S. in Computer Science; Minor in Finance
Advised by Prof. Chenren Xu
GPA: 3.83/4.00, Rank: 12/188
Work Experience
Publications
Teaching
-
CS477: Advanced Operating SystemsFall 2025, TA, EPFL
-
CS300: Data-intensive SystemsSpring 2025, TA, EPFL
-
CS477: Advanced Operating SystemsFall 2024, TA, EPFL
-
CS300: Data-intensive SystemsSpring 2024, TA, EPFL
-
CS323: Introduction to Operating SystemsFall 2023, TA, EPFL
-
CS323: Introduction to Operating SystemsFall 2022, TA, EPFL
-
04832250: Computer NetworksFall 2021, TA, Peking University
-
Turing Class: Summer Reading GroupSummer 2019, Organizer, Peking University
Awards & Honors
- Teaching Assistant Award, EDIC, EPFL, 2024
- Outstanding Graduates of Beijing, Beijing, 2022
- Outstanding Graduates of Peking University, Peking University, 2022
- China National Scholarship, China, 2021
- Merit Student, Peking University, 2021
- ASC Student Supercomputer Challenge 20-21 — 4th Place, Worldwide, 2021
- Second place, SC Student Cluster Competition 2020, Worldwide, 2020
- John Hopcroft Scholarship, Turing Class, 2020
- Schlumberger Scholarship, Peking University, 2019