Staff Research Engineer
2022 — Present
Google DeepMind
/ Gemini Thinking Team
(Cambridge, MA)
- Tech lead on latent reasoning, pre-training, and post-training research for the Gemini models.
- Developed custom attention kernels in JAX/Pallas for efficient training of novel attention mechanisms.
- Built custom internal harness/coding CLI for multi-agent power user workflows.
- 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.