I am a staff research scientist at Google DeepMind. I am broadly interested in intelligence under themes like programs and probability. Currently, I lead evaluation for Bard and also work on Gemini. My most notable works are in infrastructure (Mesh TensorFlow, Tensor2Tensor, TensorFlow Probability, Edward), modeling (Image Transformer, Automatic Differentiation Variational Inference), and evaluati