Software Engineer, TPU Compiler, PhD, Early Careers

Minimum qualifications:

• Experience with coding in data structures, algorithms and software design.
• Research experience in Artificial Intelligence, Distributed Systems, Machine Learning, Data Mining, Natural Language Processing, Image Classification, Spam Fighting, or related fields.
• Work or educational experience in Machine Learning or Artificial Intelligence.

Preferred qualifications:

• Currently enrolled in or graduated from a PhD program.
• Experience working with parallel computing.
• Experience with compilers and compiler construction.
• Excellent debugging and programming concurrent/parallel computations, and working on accelerators such as VLIW, Vector machines, GPUs, or DSPs.

About the job

TPU team develops the Accelerated Linear Algebra (XLA) TPU parallelizing compiler used to partition, optimize, and run large-scale machine learning models across multiple TPU accelerators for internal (e.g. Google DeepMind) and external customers. It is a vital part of the Google Gemini software infrastructure.

Responsibilities

• Write product or system development code for the TPU compiler (in C++).
• Participate in, or lead design reviews with peers and stakeholders to decide amongst available technologies.
• Contribute to a compiler which scales-out machine learning models across accelerators like TPU/Graphics Processing Unit (GPU) at Google and Cloud.
• Design and implement performance optimizations and critical features, which increase the velocity of important production teams.
• Apply AI to the development of the Compiler and to the Compiler itself.

Equal Employment Opportunity

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.