Staff ML Systems Architect & Co-Design Leader

Google Inc.

Sunnyvale (CA)

On-site

USD 207,000 - 300,000

Full time

14 days+

Get a reply from this employer

Your resume and cover letter, tailored to exactly what they’re hiring for.

Get past ATS filters

Benefits offered by this job

Health/dental/vision/insurance
401(k) with company match
Vacation 20 days
Sick time 40 hours/year
Maternity leave 28-30 weeks
Baby bonding leave 18 weeks
Holidays 13 paid days

Job summary

Google is seeking a Staff Software Engineer for ML Systems Co-Design in Sunnyvale, CA. You will own outcomes, influence stakeholders, and tackle ambiguous problems with deep domain expertise in ML workloads and large-scale systems.

You will lead design and implementation across the stack, drive performance modeling, and collaborate with hardware and AI research teams to advance TPU-based capabilities. This role emphasizes strong architecture, execution, and open-source model strategies.

Qualifications

  • Bachelor's degree or equivalent practical experience.
  • 8 years of experience programming in C++ or Python.
  • 5 years of experience testing and launching software products.
  • 5 years of experience with performance, large-scale systems data analysis, visualization tools, or debugging.
  • 3 years of experience with software design and architecture.

Responsibilities

  • Manage project priorities, deadlines, and deliverables.
  • Design, develop, test, deploy, maintain, and enhance software solutions.

Skills

C++
Python
Testing
Performance analysis
Large-scale systems
Debugging
Software design
Architecture

Education

Bachelor's degree or equivalent

Job description

Google is seeking a Staff Software Engineer for ML Systems Co-Design in Sunnyvale, CA. You will own outcomes, influence stakeholders, and tackle ambiguous problems with deep domain expertise in ML workloads and large-scale systems.

You will lead design and implementation across the stack, drive performance modeling, and collaborate with hardware and AI research teams to advance TPU-based capabilities. This role emphasizes strong architecture, execution, and open-source model strategies.

Get your free, confidential resume review.

or drag and drop your file here.

Similar jobs

Similar jobs worth comparing

Staff ML Compiler Performance Engineer – Co-Design Lead
Staff ML Compiler Performance Engineer – Co-Design Lead

Google LLC • Mountain View (CA)

On-site
USD 207,000 - 300,000
Senior Staff AI/ML Infrastructure Engineer
Senior Staff AI/ML Infrastructure Engineer

Google LLC • Sunnyvale (CA)

On-site
USD 262,000 - 364,000
Staff ML Systems Engineer - GPU & Performance
Staff ML Systems Engineer - GPU & Performance

Google • Sunnyvale (CA)

On-site
USD 207,000 - 300,000
Equity
Performance bonus
Staff ML Engineer - AI Systems & Leadership
Staff ML Engineer - AI Systems & Leadership

Google LLC • Sunnyvale (CA)

On-site
USD 207,000 - 300,000
Health insurance
Dental insurance
Vision insurance
+8
Staff ML Systems Architect - Co-Design & TPU Performance
Staff ML Systems Architect - Co-Design & TPU Performance

Google • Mountain View (CA)

On-site
USD 207,000 - 300,000
Staff ML Compiler Performance Engineer - Co-Design
Staff ML Compiler Performance Engineer - Co-Design

Google • Mountain View (CA)

On-site
USD 207,000 - 300,000
AI/ML Software Engineering Manager - Cloud Systems & Leadership
AI/ML Software Engineering Manager - Cloud Systems & Leadership

Google • Sunnyvale (CA)

On-site
USD 207,000 - 300,000
Performance Systems Engineer - ML & Hardware Co-Design
Performance Systems Engineer - ML & Hardware Co-Design

Google LLC • Sunnyvale (CA)

On-site
USD 147,000 - 210,000
Senior ML Engineer — Scalable AI/ML Systems Lead
Senior ML Engineer — Scalable AI/ML Systems Lead

Google LLC • San Jose (CA)

On-site
USD 174,000 - 252,000
Health, dental, vision, life, and disb
401(k) with company match
Paid time off: 20 days/year + 13 paidH
+4
Senior AI/ML Software Engineer, Cloud & TPU
Senior AI/ML Software Engineer, Cloud & TPU

Google • Sunnyvale (CA)

On-site
USD 174,000 - 252,000