The Staff Software Engineer role focuses on developing the Accelerated Linear Algebra (XLA) compiler for Google's Tensor Processing Units (TPUs), which are designed to accelerate machine learning workloads. The position involves working on a state-of-the-art TPU compiler and collaborating closely with hardware designers to co-design future processors. Engineers in this role will manage project priorities and deliverables while contributing to high-performance implementations of operations at a distributed scale. This position is part of a team that empowers Google customers with advanced AI and infrastructure capabilities.

• Bachelor's degree or equivalent practical experience.
• 8 years of experience programming in C++ or Python.
• 5 years of experience testing and launching software products.
• 5 years of experience with performance, large-scale systems data analysis, visualization tools, or debugging.
• 3 years of experience with software design and architecture.

• Contribute to the compiler for a novel processor designed to accelerate machine learning workloads.
• Target and compile high-performance implementations of operations at distributed scale.
• Design and implement new compiler passes that extract more performance out of current and next-generation TPUs.
• Collaborate closely with hardware designers to co-design future processors.
• Research high-level representations to effectively program large-scale, distributed, and heterogeneous systems.

• Employees at Google are often offered benefits like comprehensive health insurance, 401(k) matching, and flexible work arrangements, among other benefits.