Research Engineer, Gemini Large Scale Pretraining, Deepmind

Google Google · Big Tech · London, United Kingdom +2

Research Engineer role focused on large-scale pretraining of Gemini models at DeepMind. Responsibilities include enabling new research through software engineering, managing large-scale training runs, improving technical workflows, and applying engineering expertise to research challenges. Requires experience with Python-based ML libraries and machine learning, with preferred experience in large-scale systems and LLM training.

What you'd actually do

  1. Work on long term projects to enable new research, which involve multiple team members and require good software engineering and design.
  2. Respond to time-sensitive issues arising in large-scale training runs, and in ongoing research.
  3. Make systematic improvements to technical workflows and tooling used across pretraining.
  4. Apply engineering expertise to help researchers navigate technical challenges. This requires both a deep understanding of the codebase and a broad understanding of the tech stack, the model architecture and data pipeline.
  5. Develop the right balance between good engineering practices and having a flexible codebase which researchers find easy to use and extend.

Skills

Required

  • Python
  • JAX
  • Machine Learning
  • Software Engineering

Nice to have

  • large-scale system design
  • distributed systems
  • performance engineering
  • LLM training

What the JD emphasized

  • large-scale training runs
  • pretraining
  • model architecture
  • data pipeline

Other signals

  • large-scale training runs
  • pretraining
  • model architecture
  • data pipeline