There are still lots of open positions. Let's find the one that's right for you.
The Research Engineer role at Google DeepMind focuses on inference-optimized model design for Gemini pretraining, emphasizing predictable LLM quality and optimized architecture selection for fast inference. The position requires a low-level understanding of XLA primitives and practical experience with JAX code on TPUs. The ideal candidate will engage in various distillation techniques and dataset curation, contributing across the LLM preparation stack to deliver robust models.