logo

AI Job Summary

Design and build ML compiler optimizations for EdgeTPU hardware, extending frameworks like JAX and PyTorch for high-performance compilation. Debug correctness and performance issues in deployed ML models, propose compiler features to address bottlenecks, and partner with model owners on architecture decisions. Requires 5 years software development in C++ or similar, 3 years shipping products, 1 year software design, and 3 years compiler work including optimization or parallelization.

Written from this posting by Neural Jobs AI. The full description is below.

AI Resume Tailoring Sign in to use this AI Cover Letter Sign in to use this

Job Description

  • Design and build ML compiler optimizations for EdgeTPU hardware, and extend leading authoring frameworks (such as JAX and PyTorch) to enable seamless, high-performance compilation.
  • Triage, root-cause, and resolve complex correctness and performance issues encountered when deploying state-of-the-art ML models on EdgeTPU.
  • Propose, design, and implement robust compiler features and fixes to systematically address performance bottlenecks and hardware limitations.
  • Partner closely with ML model owners to influence model architectures, ensuring they are designed for optimal, efficient execution on EdgeTPU systems.
  • Own project execution from end to end, collaborating with cross-functional partners and stakeholders, and managing priorities/deadlines/deliverables for key compiler feature areas.

Minimum qualifications:

  • Bachelor’s degree or equivalent practical experience.
  • 5 years of experience with software development in one or more programming languages (e.g. C++).
  • 3 years of experience testing, maintaining, or launching software products, and 1 year of experience with software design and architecture.
  • 3 years of experience with compilers (optimization, parallelization, etc.).

Preferred qualifications:

  • Master's degree or PhD in Computer Science or a related technical field.
  • Compiler development experience in the context of accelerator-based architectures.
  • Experience working with low-level software or strong interest in developing low-level software.
  • Experience working with hardware such as CPU, GPU, or TPU.
  • Experience in optimizing ML model inference on device.
  • Bachelor’s degree or equivalent practical experience.
  • 5 years of experience with software development in one or more programming languages (e.g. C++).
  • 3 years of experience testing, maintaining, or launching software products, and 1 year of experience with software design and architecture.
  • 3 years of experience with compilers (optimization, parallelization, etc.).
Do you match this job?

Here is what this employer asked for. Sign in and we will fill in your half.

  • Role Senior Software Engineer, ML Compilers, Edge TPU
  • Experience 5-7 years
  • Education Bachelor Degree, or equivalent experience
  • Work type On-site
  • Location United States
Check my match (free)
Google
AI Research Lab · 500+ Members · Mountain View, CA, United States

Google builds internet, software, cloud, and AI products used by consumers, developers, and organizations. Its portfolio includes Search, YouTube, Android, Chrome, Maps, Gmail, Workspace, Google Cloud, advertising platforms, devices, and Gemini AI products. The company develops large-scale computing infrastructure and research that power information retrieval, communication, productivity, media, navigation, and machine learning. Google is the largest operating business within Alphabet and earns a substantial share of its revenue from digital advertising.

All jobs at Google

145 more Software Engineer roles in Mountain View

Job Overview

Approx. salary range

177K – 224K

Our estimate — this employer did not publish a salary

Our estimate, not the employer’s. Worked out from the middle half of 158 comparable roles on Neural Jobs that did publish a salary, in the same field, country and experience band. The real figure for this job may be different.

Eligibility
United States Right to work in the United States required.
Workplace
On-site
Job Posted:
3 days ago
Job Type
Full Time
Education
Bachelor Degree, or equivalent experience
Experience
5-7 years

Share This Job: