Staff ML Compiler Performance Engineer – Co-Design Lead

Google LLC; Mountain View (CA); On-site; USD 207,000 - 300,000; Full time

Job summary

Google is seeking a Staff SWE, ML Compiler Performance Engineer Co-Design in Mountain View, CA. You will tackle pre-silicon ML workloads, drive performance modeling, and contribute to RTL and compiler integration for large-scale AI systems.

You will collaborate across hardware, software, and AI research teams, own critical milestones, and deliver open-source style methodologies to map ML models to TPU performance.

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

• Lead performance engineering for ML compiler co-design projects across hardware and software.
• Develop and validate high-performance benchmarks and analysis tools.
• Collaborate across teams to drive TPU-related performance goals.

Skills

C++

Python

Software testing

Performance analysis

Software architecture

Education

Bachelor's degree or equivalent practical experience

Job description

Google is seeking a Staff SWE, ML Compiler Performance Engineer Co-Design in Mountain View, CA. You will tackle pre-silicon ML workloads, drive performance modeling, and contribute to RTL and compiler integration for large-scale AI systems.

You will collaborate across hardware, software, and AI research teams, own critical milestones, and deliver open-source style methodologies to map ML models to TPU performance.