ML Hardware Accelerators Engineer - Inference & Compiler

DeepMind Technologies Limited; Mountain View (CA); On-site; USD 181,000 - 245,000; Full time

Job summary

DeepMind Technologies Limited is seeking engineers to contribute to a next-generation AI inference platform, emphasizing memory technologies and hardware/software co-design. The role focuses on ML compiler stack and inference infrastructure, with opportunities to bridge AI workloads with hardware operators.

This coding position invites initiative-taking engineers to advance in a collaborative environment with scientists and engineers impacting AI development for Google and industry at large.

Qualifications

• Bachelor's degree or equivalent in CS or a related technical field.
• Experience with Python and C++ software development.
• Experience with ML hardware accelerators, ML compiler stacks and kernel development.

Responsibilities

• Direct full-stack Software role focusing on ML compiler and inference infrastructure.
• Design and develop ML compiler stack to bridge AI workloads and low-level HW operators.
• Design and develop software infrastructure to enable high-performance inference.
• Drive SW/HW codesign, kernel optimization, performance tuning, and debugging.

Skills

Python

C++

ML hardware accelerators

Kernel development

Education

Bachelor's degree in Computer Science or related field

PhD degree in Computer Engineering/CS

Tools

JAX

PyTorch

XLA

Job description

DeepMind Technologies Limited is seeking engineers to contribute to a next-generation AI inference platform, emphasizing memory technologies and hardware/software co-design. The role focuses on ML compiler stack and inference infrastructure, with opportunities to bridge AI workloads with hardware operators.

This coding position invites initiative-taking engineers to advance in a collaborative environment with scientists and engineers impacting AI development for Google and industry at large.