Software Engineer, Compilers and Distributed Runtime Systems

Note: By applying to this position you will have an opportunity to share your preferred working location from the following: Auckland, New Zealand; Mountain View, CA, USA; Seattle, WA, USA .

Minimum qualifications:

1. Bachelor's degree in Electrical Engineering or Computer Science, or equivalent practical experience.
2. 5 years of experience with software development and data structures/algorithms.
3. 3 years of experience testing, maintaining, or launching software products, and 1 year of experience with software design and architecture.
4. Experience with compiler technologies, distributed systems, and communication libraries (e.g. MPI).

Preferred qualifications:

1. PhD in Electrical Engineering or Computer Science.
2. Experience programming in C++ and Python.
3. Experience with compiler intermediate representations (e.g., MLIR, XLA, HLO).
4. Experience with program generation and synthesis, and behavioral simulators.
5. Experience with system-level hardware/software co-design.
6. Excellent communication, teamwork, and problem-solving skills.

About The Job:

In this role, you will develop tools to enable the co-design of ML models with both accelerators and system architecture. You'll work with an experienced team of researchers to develop compiler technologies and runtime support for modeling ML inference performance.

Responsibilities:

1. Build cost models that estimate the system performance of ML models.
2. Design and build compiler techniques for automating the partitioning and distribution of ML models.
3. Design and build runtime systems support in the OpenXLA framework to evaluate the effectiveness of above compiler transformations.

Google is proud to be an equal opportunity workplace and is an affirmative action employer. We are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity or Veteran status. We also consider qualified applicants regardless of criminal histories, consistent with legal requirements. See also Google's EEO Policy and EEO is the Law.

If you have a disability or special need that requires accommodation, please let us know by completing our Accommodations for Applicants form.