Google
Posted 5d ago

Software Engineer, ML Accelerators, DeepMind

Google
Mountain View or San Francisco
$174k-$252k/yrOnsiteFull Time
Responsibilities
  • developing compilers
  • enabling inference
  • optimizing kernels
Requirements
  • Bachelor's degree or equivalent practical experience
  • Python and C++ development
  • ML hardware accelerator
  • Compiler stack, and kernel development experience. Two years of accelerator systems experience preferred
Technical tools mentioned
PythonC++JAXPyTorchXLA

Job description

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.

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

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.

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.
Google is proud to be an equal opportunity workplace and is an affirmative action employer. We are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity or Veteran status. We also consider qualified applicants regardless of criminal histories, consistent with legal requirements. See also Google's EEO Policy and EEO is the Law. If you have a disability or special need that requires accommodation, please let us know by completing our Accommodations for Applicants form.

About Google

Global technology specializing in internet-related services and products.

Year founded
1998
Employees
181000
Organization type
Private
Latest investment
Raised $84.75B Funding Round (2026) — led by Berkshire Hathaway
Headquarters
US

Similar jobs

Software Engineer roles near Mountain View, California
9h
Save
Mark Applied
Hide
Software Engineer - Game of Thrones Slots
Austin or San Mateo or Chicago or Toronto
$71k-$127k/yr HybridFull Time
Zynga
Zynga: Mobile video game developer and publisher creating games for players worldwide as a Take-Two subsidiary.
1+ YOEB.Sc. in Computer Science or equivalent; 1–2 years software development experience, or 3+ years relevant engineering experience; Unity, C#, PHP backend, Git, algorithms, and data structures.
Unity, C#, PHP, Git
10h
Save
Mark Applied
Hide
Software Engineer, GTM
San Francisco, California, United States
$160k-$200k/yr OnsiteFull Time
Hyperbound
Hyperbound: Private AI sales-coaching platform helping enterprise revenue teams practice conversations and improve performance.
Build and own production systems for lead routing, pipeline, forecasting, scoring, deal workflows, and revenue data across CRM, product, and billing systems.
11h
Save
Mark Applied
Hide
Senior Software Engineer, AI Ecosystem 
Mountain View, California, United States
$160k-$205k/yr OnsiteFull Time
Aerospike
Aerospike: Aerospike is a private software providing a real-time database platform for enterprise AI and mission-critical applications.
5+ YOERequires 5+ years coding experience in Go, Python, Java, Scala, C, or C++, plus distributed systems integration, Kubernetes, Docker, and DevOps/SRE expertise. NoSQL, vector databases, and open-source experience are bonuses.
Go, Python, Java, Scala, C, C++, Kubernetes, Docker, NoSQL, GitHub, Stack Overflow
11h
Save
Mark Applied
Hide
Software Engineer
San Francisco, California, United States
$150k-$300k/yr OnsiteFull Time
Arini
Arini: AI receptionist and healthcare automation software provider serving dental practices and dental service organizations.
Design, deploy, and iterate AI agents and automation workflows, integrate production code with enterprise systems, and deliver measurable customer outcomes in a fast-moving environment.
EHR
1d
Save
Mark Applied
Hide
Software Engineer
Menlo Park or Seattle
$219k-$301k/yr OnsiteFull Time
Meta
MetaNASDAQ: META: Builds technologies that help people connect, find communities, and grow businesses.
10+ YOEBachelor's degree or equivalent experience, 10+ years in networking or infrastructure software, 4+ years designing production dataplane/control-plane systems, Kubernetes networking, C/C++, scripting, and test automation.
Kubernetes, C, C++, Python, Shell Scripting, DPDK, eBPF/XDP, AF_XDP, io_uring, RDMA/RoCEv2, Linux, TC/iptables/nftables
1d
Save
Mark Applied
Hide
Software Engineer Simulation Infrastructure
Cupertino or North America
OnsiteFull Time
Apple
AppleNASDAQ: AAPL: Designing and manufacturing consumer electronics, software, and digital services.
Strong Python skills, HTTP service and relational database experience, frontend framework knowledge, cloud infrastructure experience, networking and debugging expertise, Linux/macOS background, and a CS degree or equivalent experience.
Python, FastAPI, Flask, Postgres, React, Kubernetes, TCP/IP, HTTP, WebSockets, SSH, Linux, macOS, Swift, C++, OIDC, OAuth
1d
Save
Mark Applied
Hide
Software Engineer
San Francisco, California, United States
OnsiteFull Time
Proximal
Proximal: Private San Francis research lab developing coding data and benchmarks for autonomous coding agents.
Experience designing scalable systems from scratch with strong correctness, reliability, and performance; experimental instincts, systems design judgment, and ability to solve ambiguous technical problems independently.
1d
Save
Mark Applied
Hide
Software Engineer, Simulation Systems
United States or San Francisco
$140k-$185k/yr RemoteFull Time
Aalyria
Aalyria: Private aerospace communications providing network-orchestration software and optical terminals to commercial and government customers.
5+ YOEBachelor's degree in computer science or related technical field or equivalent experience; 5+ years developing production software; proficiency in Golang, C/C++, or Java; distributed or complex backend systems experience.
Golang, C, C++, Java