Back to jobs
Software Engineer, TPU Compiler Development Infrastructure
Posted 2 weeks ago
Job Description
- 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.