jobs

Staff Software Engineer, Machine Learning, TPU Workload Optimization

Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our products need to handle information at massive scale, and extend wel

Company

Google

Location

Singapore

Status

Open

Posted

2026-09-01T11:06:48+00:00

Google is hiring a Staff Software Engineer for TPU workload optimization in Singapore, on-site. The role sits in the AI Engine team, which supports serving machine learning models on Cloud GPUs and TPUs. The engineer leads onboarding of customer training and inference workloads, sets the technical direction for optimization frameworks, and prepares very large reference models, including single-host and multi-host inference. The posting asks for eight years of software development, four years leading machine learning design and infrastructure optimization, and two years with large-scale training and inference tools. The work is customer-facing.