The Research Engineer role at DeepMind focuses on inference-optimized model design for Gemini pretraining. Engineers will work on large-scale tests and deploy innovative ideas quickly, contributing to real-world problems in artificial intelligence and machine learning. The team emphasizes collaboration with universities and the wider research community, ensuring a strong connection to research roots. This position offers opportunities to advance AI development while prioritizing safety and ethics.

• Bachelor's degree in Computer Science, Mathematics, Applied Stats, Machine Learning, or a related field, or equivalent practical experience.
• 5 years of experience with inference-optimized model design.
• Experience working with product teams.

• Focus on inference-optimized model design for Gemini pretraining.
• Understand common Accelerated Linear Algebra (XLA) primitives and how JAX code runs on TPUs in practice.
• Work across the Large Language Model (LLM) preparation stack to deliver strong models.

• Employees at Google are often offered benefits like comprehensive health insurance, 401(k) matching, and flexible work arrangements, among other benefits.