Staff Software Engineer, AI Engines 3P TPU Inference

Experteer Overview In this role you will help advance ML infrastructure at scale within Google, shaping how ML systems perform across Cloud and partner products. You will collaborate with Research and cross-functional teams to drive impactful ML capabilities, including compilers and runtimes, and help migrate runtimes and workflows to ML Runtime. You will lead technical direction and deliver scalable software solutions that influence the broader Google AI stack. This is a high-impact, leadership-oriented engineering role at the intersection of ML research and production. Compensation / Benefits * Guide sustainable engineering choices for ML systems at scale * Define next directions for infrastructure over a 12-month horizon * Deliver capability and optimization improvements to Cloud and partner products * Migrate runtimes and custom workflows onto ML Runtime with minimal user disruption * Partner with GDM to transfer innovations into product teams Tasks * Bachelor’s degree or equivalent practical experience * 8 years of software development experience * 5 years testing and launching software products * 3 years in software design and architecture * 5 years of experience with ML-related domains (speech/audio, reinforcement learning, ML infrastructure, or related field) * 5 years experience with ML design and ML infrastructure (model deployment, evaluation, data processing, debugging, fine tuning) Key requirements * bonus target (20%) * equity * health benefits * career growth opportunities * work eligibility sponsorship * remote-friendly policies

Experteer Overview In this role you will help advance ML infrastructure at scale within Google, shaping how ML systems perform across Cloud and partner products. You will collaborate with Research and cross-functional teams to drive impactful ML capabilities, including compilers and runtimes, and help migrate runtimes and workflows to ML Runtime. You will lead technical direction and deliver scalable software solutions that influence the broader Google AI stack. This is a high-impact, leadership-oriented engineering role at the intersection of ML research and production. Compensation / Benefits * Guide sustainable engineering choices for ML systems at scale * Define next directions for infrastructure over a 12-month horizon * Deliver capability and optimization improvements to Cloud and partner products * Migrate runtimes and custom workflows onto ML Runtime with minimal user disruption * Partner with GDM to transfer innovations into product teams Tasks * Bachelor’s degree or aaaaaa help practical experience * 8 years of software development experience * 5 years testing and launching software products * 3 years in software design and architecture * 5 years of experience with ML-related domains (speech/audio, reinforcement learning, ML infrastructure, or related field) * 5 years experience with ML design and ML infrastructure (model deployment, evaluation, data processing, debugging, fine tuning) Key requirements * bonus target (20%) * equity * health benefits * career growth opportunities * work eligibility sponsorship * remote-friendly policies