New! May 2020: I started a new research position at Google Brain to study applications of ML to high-performance systems.
New! Mar 2020: We presented optimal rematerialization system, Checkmate, at MLSys 2020.
Oct 2019: We're presenting a system for checkpointing for DNNs to enable training enourmous DNNs on a single GPU.
Jan 2019: fiJIT is a high-performance system for DNN GPU multiplexing
Dec 2018: Revec: Program Rejuvenation through Revectorization was accepted to Compiler Construction 2019!
November 2018: Dynamic Space-Time Scheduling for GPU Inference was accepted to LearningSys (at NeurIPS 2018)!
I’m a Ph.D. student at UC Berkeley. As a Ph.D. student, I am advised by Ion Stoica and Joseph Gonzalez. I am a part of the RISE Lab, BAIR Lab and Berkeley Deep Drive Lab. I previously was a Research Scientist at DeepScale, a startup building tiny deep learning systems for autonomous cars.