I am an associate professor in the Department of Computer Science at ETH Zurich and a part of the Systems group. Prior to this I was a faculty member at UW-Madison and a part of the madSystems group. I completed my PhD from UC Berkeley where I was advised by Ion Stoica and Mike Franklin. I also have a Masters from University of Illinois at Urbana-Champaign.
Current research areas
My group works at the intersection of computer systems and machine learning, in both directions. On one side, we work across the stack to make modern ML workloads faster, cheaper, and more reliable: from efficient ML model architecture design to GPU energy modeling. On the other, we use machine learning inside core systems, in areas such as memory tiering and cloud configuration tuning.
- Improving LLM inference: Reducing the cost of running large language models through inference-efficient model architectures, memory management, and speculative decoding.
- GPU variability & power management: Understanding how variability across GPUs affects cluster performance through variability-aware scheduling and high-fidelity GPU energy modeling.
- Vector search: New indexing and query methods to make vector similarity search faster and more scalable, including adaptive indexes and vector database deployment on HPC platforms.
- Integrating ML into systems: Using machine learning to improve core system components, including memory tiering and tuning unstable and noisy cloud applications.
Group
- Rutwik Jain — co-advised w/ Matt Sinclair
- Brandon Tran — co-advised w/ Matt Sinclair
- Minghao Yan
- Johannes Freischuetz
- Tzu-Tao Chang
- Fanchao Chen
- Tareq Mahmood
- Seth Ockerman
Alumni
+PhD
- Song Bian → NVIDIA Research Labs
- Konstantinos Kanellis → AWS Learned Systems Group
- Jason Mohoney → Post-doc at MIT
- Saurabh Agarwal → Post-doc at UT-Austin
Post-doctoral researchers
- Pengfei Zheng (co-advised with Aditya Akella) → Huawei Technologies
MS
- Devesh Sarda → Databricks
- Aditi Singh → Nutanix
- Mohil Patel → Oracle
- Rachit Tibrewal
- Olesia Elfimova → Dropbox
- Adarsh Kumar → Amazon Alexa AI
- Arjun Balasubramanian → Amazon AWS
Selected recent publications
Please see Google Scholar for a complete list.