Awards
- Google ML and Systems Junior Faculty Awards, 2025.
 - Google Research Scholar, 2025.
 - Conference on Machine Learning and Systems (MLSys), Outstanding Paper Honorable Mention, 2025.
 - Conference on Language Modeling (COLM), Outstanding Paper, 2024.
 - Inaugural Stanford Open Source Software Prize for Flash Attention, 2024.
 - International Conference on Machine Learning (ICML), Outstanding Paper runner-up, 2022.
 - ICML Hardware Aware Efficient Training Workshop, Best Paper award, 2022.