Software Engineer, TPU Compiler Development Infrastructure
Sunnyvale, CA
Full-time
Onsite
$147K/yr - $211K/yr
Entry Level
Google is looking for a Software Engineer to join their TPU Compiler Development Infrastructure team. The role focuses on improving infrastructure to enhance developer productivity and reduce latency in the compiler workflow for TPUs, Google's custom-designed processors for machine learning workloads.
Reduce CL time to submit for a CL and minimize CL rollback for the whole XLA TPU team
Drive infrastructure improvements that remove friction from the daily development of the XLA TPU Compiler team
Develop tools supporting compiler engineers as they work through stages of new TPU introduction (e.g., testing when hardware is not yet available or very limited)
Modernize and simplify build/test fixtures (e.g. xla_test) to make them more reliable and easier for the team to use
Design and implement system architectures which cleanly handle ever increasing number of TPU generations and compiler features, ensuring the codebase doesn't become a "spaghetti" of special cases
Identify and resolve accelerator utilization bottlenecks, improve accelerator test coverage without slowing down CL submission
Qualification
Required
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
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
Benefits
Bonus
Equity
Benefits
Google specializes in internet-related services and products, including search, advertising, and software. It is a sub-organization of Alphabet.
