Requirements

• Experience in multidisciplinary research, guiding, and mentoring.
• Machine Learning experience.

Nice to haves

• PhD in Computer Science with a strong publication history.

What you'll be doing

• Contribute to the creation of novel multi-agent learning algorithms and frameworks, focusing on performance and scalability in JAX.
• Build and maintain simulation platforms and end-to-end research pipelines for experiments on DeepMind’s cutting-edge infrastructure, including massive TPU pods.
• Partner with Research Scientists to translate mathematical concepts and research hypotheses into robust, production-quality code and reproducible experiments.
• Lead engineering direction for complex research projects, establish best practices for code quality and maintainability, and mentor junior engineers.
• Optimize the research workflow from data processing and model training to results analysis, accelerating the pace of discovery.

Perks and benefits

• Opportunities to work on real-world problems in AI, data mining, natural language processing, hardware and software performance, compilers, and core search.
• Engagement in a pioneering AI lab focused on advancing AI for global challenges and product innovation benefiting billions of users.
• Collaboration with interdisciplinary teams and partner universities worldwide.
• Focus on safety, ethics, and public benefit in AI technology development.
• Diverse learning opportunities and varied career pathways through global team collaboration.