ML Inference & Compiler Engineer - HW/SW Codesign

DeepMind Technologies Limited; San Francisco (CA); Híbrido; USD 174.000 - 252.000; Jornada completa

Descripción de la vacante

DeepMind is hiring engineers to build a next-generation compute platform for AI inference workloads, focusing on memory technologies, hardware acceleration, and SW/HW codesign. This coding role emphasizes ML compiler stack and inference infrastructure, joining a team of scientists and engineers who impact the future of machine learning for Google and industry.

As part of DeepMind, you will work on bridging AI workloads to low-level hardware operators, drive high-performance inference, and

Formación

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

Responsabilidades

• 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 SW infrastructure to enable high performance inference.
• Drive SW/HW codesign, kernel optimization, performance tuning, and debugging.

Conocimientos

Python

C++

ML accelerators

ML compiler stacks

Kernel development

Educación

Bachelor's degree in Computer Science or related field

PhD in Computer Science / Computer Engineering

Herramientas

JAX

PyTorch

XLA

Descripción del empleo

DeepMind is hiring engineers to build a next-generation compute platform for AI inference workloads, focusing on memory technologies, hardware acceleration, and SW/HW codesign. This coding role emphasizes ML compiler stack and inference infrastructure, joining a team of scientists and engineers who impact the future of machine learning for Google and industry.

As part of DeepMind, you will work on bridging AI workloads to low-level hardware operators, drive high-performance inference, and