Software Engineer, ML Accelerators, DeepMind

Google Inc.; Mountain View, San Francisco (CA, CA); On-site; USD 174,000 - 252,000; Full time

Job summary

DeepMind in Mountain View is seeking a Software Engineer focused on ML accelerators and the ML compiler stack to drive inference infrastructure. This is a hands-on role for engineers who take initiative and deliver results, bridging AI workloads with low-level hardware operators.

You will collaborate with scientists and engineers across Google DeepMind to build a next-generation compute platform for AI inference, with emphasis on memory technologies, hardware acceleration, and software/hardware

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.

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

Skills

Python

C++

ML accelerators

ML compiler stacks

Kernel development

Education

Bachelor's degree in Computer Science or related field

Tools

JAX

PyTorch

XLA

Job description

Software Engineer, ML Accelerators, DeepMind

Share Software Engineer, ML Accelerators, DeepMind

• link Copy link

DeepMind Mountain View, CA, USA ; San Francisco, CA, USA

Share Software Engineer, ML Accelerators, DeepMind

• link Copy link

X Applicants in San Francisco: Qualified applications with arrest or conviction records will be considered for employment in accordance with the San Francisco Fair Chance Ordinance for Employers and the California Fair Chance Act. Note: By applying to this position you will have an opportunity to share your preferred working location from the following: Mountain View, CA, USA; San Francisco, CA, USA.

• 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.

Preferred qualifications:

• PhD degree in Computer Engineering, Computer Science, or a related field.
• 2 years of experience with full-stack system development for hardware accelerators.
• 2 years of experience with collaborative, hyper-scalar software development and engineering best practices.
• Experience scoping, planning, and executing projects towards team goals in ambiguous environments.

About the job

We are building a next-generation compute platform for AI inference workloads, with an emphasis on memory technologies, Hardware (HW) acceleration, and Software (SW)/HW codesign.

We are looking for enthusiastic engineers to join the project.

This is a coding role for engineers who like to take initiative and get things done, with a focus on ML compiler stack and inference infrastructure.

At DeepMind you will be joining our team of scientists and engineers who are directly impacting the future of machine learning for Google and industry.

Artificial intelligence will be one of humanity’s most transformative inventions. At Google DeepMind, we are a pioneering AI lab with exceptional interdisciplinary teams focused on advancing AI development to solve complex global challenges and accelerate high-quality product innovation for billions of users. We use our technologies for widespread public benefit and scientific discovery, ensuring safety and ethics are always our highest priority.

We are pushing the boundaries across multiple domains. Our global teams offer diverse learning opportunities and varied career pathways for those driven to achieve exceptional results through collective effort. Individual pay is determined by factors including job-related skills, experience, and relevant education or training.

US: $174000 - $252000 (USD) + 15% bonus target + equity + benefits

Learn more about benefits at Google.

• 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.

Google is proud to be an equal opportunity and affirmative action employer. We are committed to building a workforce that is representative of the users we serve, creating a culture of belonging, and providing an equal employment opportunity regardless of race, creed, color, religion, gender, sexual orientation, gender identity/expression, national origin, disability, age, genetic information, veteran status, marital status, pregnancy or related condition (including breastfeeding), expecting or parents-to-be, criminal histories consistent with legal requirements, or any other basis protected by law. See also Google's EEO Policy , Know your rights: workplace discrimination is illegal , Belonging at Google , and How we hire .

Google is a global company and, in order to facilitate efficient collaboration and communication globally, English proficiency is a requirement for all roles unless stated otherwise in the job posting.

To all recruitment agencies: Google does not accept agency resumes. Please do not forward resumes to our jobs alias, Google employees, or any other organization location. Google is not responsible for any fees related to unsolicited resumes.

Equity is granted exclusively and discretionarily by Alphabet Inc. on the basis of an agreement concluded between you and Alphabet Inc. Alphabet Inc. is your sole contractual partner with respect to equity grants. GSU grants are not guaranteed, are discretionary, are subject to approval by the Alphabet Inc. board of directors or its delegate, the terms of the relevant Alphabet Inc. stock plan, and your grant agreement. They have no impact on statutory payments. Current or past grants do not confer an acquired right.