The Staff Software Engineer role focuses on AI/ML infrastructure for TPU supercomputers at Google. The position involves leading the design and delivery of specialized AI compute platforms while ensuring high availability for massive-scale workloads. The Emergent AI infrastructure team aims to redefine capabilities in AI and infrastructure, empowering customers with advanced solutions. This role requires collaboration with hardware engineering and chip design teams to influence future architectures for large-scale deployment.

• Bachelor's degree or equivalent practical experience.
• 8 years of software development experience in C, C++, Go, or Python.
• 5 years of experience testing and launching software products.
• 5 years of experience building and developing large-scale infrastructure, distributed systems, or networks.
• 3 years of experience designing, building, and operating large-scale distributed systems.

• Lead the design and end-to-end software delivery of specialized AI compute platforms.
• Establish observability and hardening strategies for the lower-half software stack.
• Architect robust integration interfaces between custom compute topologies and industry-standard workload schedulers.
• Architect secure boot and cryptographic remote attestation flows for distributed hardware platforms.
• Partner closely with hardware engineering and chip design teams to influence future architectures.

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