Hey, I'm Arjun!

I'm currently a research engineer at Google DeepMind working on Gemini pretraining architecture and performance.

Previously, I was working on the XLA compiler for TPU on kernels & compiler optimizations, and before that I was an undergrad at UC Berkeley.

Catch me on twitter!