Your browser does not support javascript! Please enable it, otherwise web will not work for you.

Senior Software Engineer, ML Compiler

Home > Python programming jobs

Senior Software Engineer, ML Compiler in USA

  • Sunnyvale, CA

Responsibilities

  • Deliver compiler parallelization features and optimization techniques for TPU backends supporting large-scale workloads.
  • Contribute to collective operation lowering and implementation on the TPU platform.
  • Develop low-level compiler optimization techniques throughout the compiler stack.
  • Analyze TPU architecture features and use them to optimize horizontal scaling performance.
  • Build compiler debugging tools to prevent scaling issues and improve engineering experience.

Requirements

  • Bachelor's degree in Computer Science or a related technical field, or equivalent practical experience.
  • At least 5 years of software development experience in one or more programming languages, including Python and C++.
  • At least 3 years of experience with machine learning infrastructure, ML execution frameworks such as TensorFlow, JAX, or PyTorch, or hardware accelerators such as TPUs or GPUs.
  • At least 2 years of experience with a low-level systems programming language such as C++.
  • Experience with performance and compilers.
  • Preferred: 3 years of experience in low-level ML accelerator programming, compiler programming, or close-to-hardware performance programming.
  • Preferred: experience profiling workloads and implementing performance optimizations.
  • Preferred: experience writing high-performance, readable C++.
  • Preferred: experience with hardware design and hardware architecture.
  • Preferred: working knowledge of ML compilers such as XLA or MLIR and experience co-designing hardware-aware optimizations for model execution.

Benefits

  • 15% bonus target, equity, and benefits are offered in addition to base pay.

Salary: $174k - $252k/yr

Google

A problem isn't truly solved until it's solved for all. Googlers build products that help create opportunities for everyone, whether down the street or across the globe. Bring your insight, imagination and a healthy disregard for the impossible. Bring everything that makes you unique. Together, w...

Similar positions

Software Engineer in Test

  • Roadie
  • Full time
  • USA
  • 09/13/2026
  • Salary: Competitive
  • Remote

Principal Software Engineer - Builder Experience

  • Elastic
  • Full time
  • Remote
  • 09/13/2026
  • Salary: €73k-€116k
  • Greece

Principal Software Engineer - Builder Experience

  • Elastic
  • Full time
  • Remote
  • 09/13/2026
  • Salary: zł 369k-zł 584k
  • Poland

Senior Devops Engineer

  • Woliba
  • Full time
  • USA
  • 09/13/2026
  • Remote

Senior Core Infrastructure Engineer

  • Oracle
  • Full time
  • USA
  • 09/13/2026
  • Salary: Competitive
  • Nashville, TN

Listings

by


Job Type

Language

Country

CodingJobBoard.com © 2026