Software Engineer, ML Accelerators, DeepMind
Google· DeepMind
Hybrid (Mountain View, CA; San Francisco, CA)$174k - $252kFull-timePosted 1w ago
About the role
Engineering role focused on ML compiler stack and inference infrastructure for AI acceleration, with emphasis on HW acceleration and SW/HW codesign, contributing to next-generation compute platform for AI inference workloads.
Responsibilities
- Direct full-stack Software (SW) role, focusing on ML compiler and inference infrastructure.
- Design and develop ML compiler stack to bridge AI workloads and low-level Hardware (HW) operators.
- Design and develop SW infrastructure to enable high performance inference.
- Drive SW/HW codesign, kernel optimization, performance tuning, and debugging.
Qualifications
- Bachelor's degree in Computer Science, a related technical field, or equivalent practical experience.
- Experience in software development using Python and C++.
- Experience with machine learning (ML) hardware accelerators, ML compiler stacks (e.g., JAX, PyTorch, XLA), and kernel development.
Skills
Required:PythonC++ML hardware acceleratorsML compiler stackskernel development
Benefits
- Learn more about benefits at Google (link provided in description)
Last updated Oct 23, 2026. Interested? Apply on the company site.
More open roles at Google
- Staff Network Security Engineer, Google Enterprise Network3d agoHybrid (Austin, TX; Sunnyvale, CA)
- Group Product Manager, Google Lens AI Shopping3d agoMountain View, CA, USA · Product Management
- UX Program Manager3d agoHybrid (New York, NY; San Bruno, CA) · UX Programs & Operations
- Senior Staff Software Engineer, Android Automotive OS3d agoHybrid (Kraków, Poland; Munich, Germany)
- 3d ago