Research Engineer, Multi Agent Learning, DeepMind

Requirements:

• Bachelor’s degree in Computer Science or a related technical field, or equivalent practical experience.
• 5 years of experience in software development using Python or C++.
• Experience in deep learning frameworks, such as JAX or PyTorch.
• Experience in standard analysis and scientific computing libraries such as numpy, pandas, and matplotlib.

Nice to haves:

• PhD with a research focus on Machine Learning, Reinforcement Learning, or Multi-Agent Systems.
• Experience training large-scale models on accelerators (TPUs, GPUs) in a distributed environment.
• Experience working with language models such as designing agentic harnesses, memory retrieval, or fine-tuning.
• A track record of leading complex software projects and a passion for enabling groundbreaking research through engineering.
• Deep expertise building and optimizing complex systems in JAX.
• A strong background in multi-agent reinforcement learning, algorithmic game theory, or computational economics.

What you'll be doing:

• Contribute to the creation of novel multi-agent learning algorithms and frameworks, with a focus on performance and scalability in Just-In-Time (JIT) compilation, Autograd, and XLA (JAX).
• Build and maintain large-scale simulation platforms and end-to-end research pipelines to run experiments on Google DeepMind’s cutting-edge infrastructure, including massive TPU pods.
• Partner deeply with Research Scientists to transform mathematical concepts and research hypotheses into robust, production-quality code and reproducible experiments.
• Lead the engineering direction for complex research projects, establish best practices for code quality and maintainability, and mentor junior engineers on the team.
• Optimize every part of the research workflow, from data processing and model training to results analysis, to accelerate the pace of discovery.

Perks and benefits:

• Opportunity to work at Google DeepMind, a pioneering AI lab with interdisciplinary teams working on solving complex global challenges.
• Engage in research-focused software engineering to contribute to groundbreaking advancements in the field of artificial intelligence.
• Collaborate with research programs at partner universities and technical institutes around the world.
• Varied career pathways and learning opportunities within global teams at DeepMind.