Software Engineer, TPU Compiler Development Infrastructure

Job description

Minimum qualifications: - Bachelor’s degree or equivalent practical experience. - 2 years of experience with coding in C++ and Python, or 1 year of experience with an advanced degree. - 2 years of experience working with Google Infrastructure such as Blaze, TAP, or Guitar. Preferred qualifications: - Master's degree or PhD in Computer Science, or a related technical field. - Interest in becoming an expert in infrastructure surrounding low-level ML hardware programming. About the job: Our team develops the Accelerated Linear Algebra (XLA) compiler which enables TPUs, Google's in-house custom designed processor, to accelerate machine learning and other scientific computing workloads for both internal Google customers and external Cloud customers. The XLA TPU team is reaching a critical threshold of complexity at a time when the demand for rapid iteration has never been higher. This role is designed to manage the infrastructure friction that compiler engineers face daily, effectively multiplying output of the entire team. In concrete terms we need to pull down the average team presubmit latency from the current 1.5. hours to 20 min and minimize Changelist (CL) rollback (catch issues e...