Research Engineer, Pretraining, DeepMind

At Google, research-focused Software Engineers are embedded throughout the company, allowing them to set up large-scale tests and deploy promising ideas quickly and broadly. Ideas may come from internal projects as well as collaborations with research programs at partner universities and technical institutes worldwide. Responsibilities include artificial intelligence, data mining, natural language processing, hardware and software performance analysis, improving compilers for mobile platforms, core search, and more. Engineers remain connected to their research roots by partnering with universities and publishing papers.

At Google DeepMind, we are a pioneering AI lab with interdisciplinary teams advancing AI development to solve complex global challenges and accelerate high-quality product innovation for billions of users. Our technologies are used for public benefit, scientific discovery, and ethical AI development. We push boundaries across multiple domains, offering diverse learning opportunities and career pathways for those driven to achieve exceptional results through collective effort.

Minimum Qualifications

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

Preferred Qualifications

• Experience conducting applied research to improve the quality and training/serving efficiency of large transformer-based models.
• Experience with fine-tuning large models (e.g., supervised, RLHF).

Responsibilities

• 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 (e.g., pretraining, fine-tuning, serving) to deliver strong models.

Pay

Individual pay is determined by factors including job-related skills, experience, and relevant education or training. US: $174,000 - $253,000 (USD) + 15% bonus target + equity + benefits.

Benefits

Learn more about benefits at Google.