The Staff Software Engineer for the TPU Machine Learning Supercomputer at Google will design and develop features to enhance the scalability and reliability of large-scale software across TPUs and distributed hardware. This role is part of the AI and Infrastructure team, which focuses on delivering advanced AI capabilities and infrastructure at scale. The engineer will work on critical system software that supports TPU accelerators and will provide leadership in software development for future supercomputer generations. This position requires versatility and a proactive approach to problem-solving in a fast-paced environment.

• Bachelor's degree or equivalent practical experience.
• 8 years of experience with software development in C++ or Go.
• 5 years of experience with large-scale infrastructure, distributed systems, or networks.
• 3 years of experience with software design and architecture.
• Experience developing, integrating, and testing system and user-space software for hardware accelerators or TPU systems.

• Design, develop, test, deploy, and debug critical system software for TPU Machine Learning accelerators.
• Develop advanced analytics and health management capabilities for large-scale ML systems.
• Lead high-impact projects and ensure successful delivery aligned with team strategies.
• Provide technical guidance and mentorship to software engineers.

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