Senior ML Compiler Engineer for TPU Backends

Google Inc.; Sunnyvale (CA); On-site; USD 174,000 - 252,000; Full time

Job summary

Google Inc. Sunnyvale, CA, seeks a Senior Software Engineer to advance ML compiler work for TPU backends, driving performance and scalable execution.

You will collaborate with ML, systems, and hardware teams to implement compiler optimizations, analyze TPU architectures, and build tools that prevent scaling issues. Responsibilities include delivering parallelization features and mentoring junior engineers.

Qualifications

• BS in CS or related field or equivalent experience.
• 5 years of software development in Python and C++.
• 3 years in ML infra, ML frameworks or hardware accelerators.
• 2 years in low-level systems programming (C++).
• Experience with performance and compiler topics.

Responsibilities

• Deliver compiler parallelization features and optimization techniques for TPU backend.
• Contribute to collective operation lowering/implementation on TPU platform.
• Develop compiler optimization techniques across the compiler stack.
• Analyze TPU architectures to enable efficient horizontal scaling.
• Build compiler tools for debugging and preventing scaling issues.

Skills

Python

C++

ML frameworks

Performance tuning

Compiler know-how

Education

Bachelor's degree in CS or related field

Tools

TensorFlow

JAX

PyTorch

TPUs

GPUs

Job description

Google Inc. Sunnyvale, CA, seeks a Senior Software Engineer to advance ML compiler work for TPU backends, driving performance and scalable execution.

You will collaborate with ML, systems, and hardware teams to implement compiler optimizations, analyze TPU architectures, and build tools that prevent scaling issues. Responsibilities include delivering parallelization features and mentoring junior engineers.