Software Engineer (AI)

TL;DR

Software Engineer (AI): Developing and optimizing compiler infrastructure for generative machine learning and silicon automation with an accent on MLIR, LLVM, and differentiable programming. Focus on building high-performance systems for generative models and reinforcement learning to accelerate AI hardware development.

Company

DeepMind is a world-leading research organization dedicated to solving intelligence to advance science and benefit humanity.

What you will do

• Build and maintain MLIR-based compiler infrastructure for AI hardware.
• Implement generative machine learning and reinforcement learning algorithms.
• Apply formal methods to verify complex computing systems.
• Develop differentiable programming frameworks.
• Write high-quality production code for system development.
• Participate in design reviews to ensure architectural integrity.

Requirements

• Strong proficiency in C++ and Python.
• Experience with compiler construction, specifically LLVM or MLIR.
• Deep understanding of machine learning frameworks and generative modeling.
• Knowledge of formal methods for system verification.
• Ability to work on low-level machine learning acceleration.