Staff Research Engineer
2022 — Present
Google DeepMind
/ Gemini Thinking Team
(Cambridge, MA)
- Tech lead on latent reasoning and pretraining research for the Gemini models.
- Developed custom attention kernels in JAX/Pallas for efficient transformer training.
- Designed agent workflows and harnesses for mathematical and code reasoning.
- Led the Muse project (ICML 2023): delivered Google's first generative image product; the novel editing algorithm was featured at the Google I/O 2023 keynote and launched on Vertex AI.
- Mentored two MIT PhD researchers; inventor on four filed patents.