Research Engineer, Gemini Diffusion

Snapshot

Artificial Intelligence could be one of humanity’s most useful inventions. At Google DeepMind, we’re a team of scientists, engineers, machine learning experts and more, working together to advance the state of the art in artificial intelligence. We use our technologies for widespread public benefit and scientific discovery, and collaborate with others on critical challenges, ensuring safety and ethics are the highest priority.

About Us

We’re looking for a Research Engineer to help accelerate our research in Artificial Intelligence as part of our mission to create the next generation of Generative AI technologies. Our team has a unique mandate: to work on speculative, longer-horizon research projects that, if successful, yield step-change improvements in the capabilities of our frontier models. We target paradigm-shifting ideas and are prepared to invest the time and resources necessary to achieve them. Our goal is to ideate, research, develop, validate, and integrate these improvements into major Google technologies.

Our primary research focus is Gemini Diffusion, our state-of-the-art text diffusion model. We believe that text diffusion has the potential to unlock a new paradigm in generative AI, offering unprecedented capabilities and lightning-fast latency. With strong backing from GDM leadership, our team is dedicated to thoroughly exploring and advancing this technology.

Our team is a multi-disciplinary group of engineers and scientists that work in close collaboration, leveraging our unique skills with a strong focus on building and shipping ground-breaking AI. We are a problem-focused team, meaning we are interested in any tool or idea that can solve a concrete problem, rather than being tied to a particular set of tools.

The Role

We are looking for a Research Engineer to join our team in London and help us accelerate our mission. As a Research Engineer, you will apply your engineering and research skills to advance paradigm-shifting research at a large scale. You will be at the heart of our efforts to deliver step-changes in the capabilities of our frontier models, with a significant focus on our Gemini Diffusion project. You will work in a tight-knit and highly collaborative team environment that fosters both engineering and research skills, giving you hands-on experience with improving state-of-the-art AI systems. As a team member, your work may involve:

Key responsibilities:

• Brainstorming new disruptive ideas that could become the next generation of frontier AI models, particularly within the text diffusion space.
• Building and developing these ideas with the rest of the team, contributing directly to Gemini Diffusion research.
• Solving key research challenges by designing and executing experimental research on text diffusion models, sharing analyses, and proposing next steps.
• Rigorously validating the performance impact of our work at large scale.
• Working with other Generative AI teams to move the technologies we develop out of the lab and into production.
• Performance engineering, benchmarking, and optimisation, especially for large-scale diffusion models.
• Bringing engineering expertise into our projects. You will share your skills and knowledge with other engineers and researchers. We take engineering standards seriously, and you will be a steward of that ethos.

About You

In order to set you up for success as a Research Engineer at Google DeepMind,  we look for the following skills and experience:

• Degree in computer science, electrical engineering, science, mathematics, or equivalent experience.
• Programming experience, particularly with Python-based scientific libraries such as Numpy, Scipy, JAX, PyTorch, or TensorFlow.
• Familiarity with the machine learning, mathematics, and statistics needed to follow relevant research papers (linear algebra, calculus, etc.) and reason about cutting-edge innovations.
• A track record of building software, either in open source or as part of a company product or research paper.
• Agile and solution-focused mindset - be prepared to work on anything!

In addition, the following would be an advantage: 

• Academic research experience in machine learning, publications, or research experience in related fields.
• Strong communication skills (via discussion, presentation, technical and research writing, whiteboarding, etc.).
• Large-scale system design, distributed systems.
• Distributed computation for ML, especially in the context of accelerators (e.g., sharding, multi-host computation).
• C++ or broader programming experience.
• Data engineering and visualisation.

At Google DeepMind, we value diversity of experience, knowledge, backgrounds and perspectives and harness these qualities to create extraordinary impact. We are committed to equal employment opportunity regardless of sex, race, religion or belief, ethnic or national origin, disability, age, citizenship, marital, domestic or civil partnership status, sexual orientation, gender identity, pregnancy, or related condition (including breastfeeding) or any other basis as protected by applicable law. If you have a disability or additional need that requires accommodation, please do not hesitate to let us know.

Applications close on Friday 18th July 2025.