The Staff Software Engineer for AI Engines 3P TPU Inference at Google will focus on developing next-generation technologies that enhance machine learning infrastructure. This role requires deep expertise in ML compilers and runtimes, as well as strong leadership skills to manage project priorities and deliverables. The engineer will work closely with cross-organizational teams to drive impactful innovations and optimize existing frameworks. The position is critical to supporting Google's mission of advancing AI across various industries.

• Bachelor’s degree or equivalent practical experience.
• 8 years of experience in software development.
• 5 years of experience testing and launching software products.
• 3 years of experience with software design and architecture.
• 5 years of experience with ML design and ML infrastructure.

• Exercise judgment to guide sustainable engineering choices for ML systems at scale.
• Innovate next directions for infrastructure over a 12-month time horizon.
• Deliver impactful capability and optimization impact to Cloud and partner product areas.
• Gracefully migrate existing frameworks and runtimes onto ML Runtime.
• Partner with GDM to transfer key innovations into products developed by your team.

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