Software Engineer - TPU Compiler (PhD)

Position Overview

Join Google as a Software Engineer focused on the TPU Compiler, contributing to cutting-edge technologies that impact billions globally. This role centers on developing and optimizing the Accelerated Linear Algebra (XLA) compiler for TPU accelerators used extensively within Google and by Cloud customers. Our engineers thrive in a dynamic environment involving scalable system design, distributed computing, and artificial intelligence advancements.

Key Responsibilities

• Develop core system and product code for the TPU compiler primarily in C++.
• Engage actively in design reviews with colleagues and stakeholders to evaluate and select effective technological solutions.
• Enhance the compiler to efficiently partition and optimize large-scale machine learning models across TPU and GPU accelerators.
• Implement performance tuning and critical features that accelerate production team workflows.
• Integrate artificial intelligence techniques to improve both the compiler development process and the compiler itself.

Required Qualifications

• Proven experience writing code involving data structures, algorithms, and software design principles.
• Research background in AI, distributed systems, machine learning, data mining, natural language processing, image recognition, spam detection, or similar domains.
• Practical work or academic experience in the fields of Machine Learning or Artificial Intelligence.

Preferred Qualifications

• Currently enrolled in or graduated from a PhD program.
• Hands-on familiarity with parallel computing architectures.
• Experience working with compiler technologies and constructing compilers.
• Strong debugging skills and proficiency programming concurrent and parallel computations on accelerators such as VLIW, vector machines, GPUs, or DSPs.

Additional Information

Interviews for this position typically require in-person attendance.

Google is an equal opportunity employer, celebrating diversity and committed to equitable employment choice regardless of race, color, religion, sex, national origin, sexual orientation, age, disability, gender identity, veteran status, or other protected characteristics. We accommodate applicants with disabilities upon request.