Google

Software Engineer, ML Accelerators, DeepMind

Google$174K — $252K *
Consumer Technology
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • Bachelor's degree in Computer Science or related field, or equivalent experience.
  • Experience in Python and C software development.
  • Hands-on experience with ML hardware accelerators and compiler stacks (e.g., JAX, PyTorch, XLA).
  • PhD in Computer Engineering, Computer Science, or a related field preferred.
  • 2 years of full-stack system development experience for hardware accelerators.
  • 2 years of experience in collaborative, hyper-scalar software development.
  • Experience in scoping and executing projects in ambiguous environments.

Responsibilities

  • Perform full-stack software development, focusing on ML compiler and inference infrastructure.
  • Design and build an ML compiler stack to connect AI workloads with low-level hardware operations.
  • Create software infrastructure that supports high-performance inference.
  • Engage in software and hardware co-design, kernel optimization, and performance tuning.
  • Debug and enhance performance components.

Benefits

  • Access to comprehensive health benefits and wellness programs.
  • Opportunities for professional development and collaboration with leading scientists.
  • Flexible working locations in Mountain View, CA or San Francisco, CA.
  • 15% bonus target and potential equity options.
Full Job Description
info_outline
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.

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.

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.


About Google

Google is a multinational technology company that specializes in Internet-related services and products. These include online advertising technologies, search engine, cloud computing, software, and hardware. Google was founded in 1998 by Larry Page and Sergey Brin while they were Ph.D. students at Stanford University. The company has grown tremendously since then and has become one of the most valuable companies in the world. Google's mission is to organize the world's information and make it universally accessible and useful.
Learn more about Google
Size
156,500 employees
Market Cap
$1,115.4 billion
Industry
Net Income
$40.2 billion
Founded
1998
5 Year Trend
+23.3%
Revenue
$182.5 billion
NASDAQ

Similar Jobs

More Jobs at Google

More Consumer Technology Jobs

Find similar Software Engineer, ML Accelerators, DeepMind jobs: