Drimva

Google

Software Engineer, ML Accelerators, DeepMind

Mountain View, CA, USA +1 · Full-time · Posted Oct 14, 2026

Location
Mountain View, CA, USA
San Francisco, CA, USA
Experience
No overall years specified; preferred qualifications include 2 years of experience in each of two areas.
Education
Bachelor's degree in Computer Science, a related technical field, or equivalent practical experience; PhD preferred.

About the job

Develop software for ML compiler and inference infrastructure, connecting AI workloads to hardware and optimizing performance through software/hardware codesign.

What you'll do

  • Design and develop an ML compiler stack that bridges AI workloads and low-level hardware operators.
  • Build software infrastructure for high-performance inference.
  • Drive software/hardware codesign, kernel optimization, performance tuning, and debugging.

What they ask for

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

Nice to have

  • PhD in Computer Engineering, Computer Science, or a related field.
  • Two years of experience with full-stack hardware-accelerator system development or collaborative, hyper-scalar software development and engineering best practices.

Pay

USD $174000 - $252000 (base salary)

Tools

Python, C++, JAX, PyTorch, XLA

AI focus

Build ML compiler and inference infrastructure for AI workloads, including hardware acceleration, kernel optimization, and software/hardware codesign.

Model-written summary · the original posting is authoritative

First seen Oct 21, 2026 · Last verified Nov 4, 2026 (UTC)