Back to jobs
Research Engineer, Multi Agent Learning, DeepMind
Posted 3 days ago
Job Description
- 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.