jobs

Senior 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

In inglese

Company

Google

Location

Singapore

Status

Open

Posted

2026-09-02T07:40:02+00:00

A senior software engineering role on Google's AI and Infrastructure team, part of AI Engine, based on-site in Singapore. The engineer leads onboarding for a stable stack and optimization of training and serving on the latest TPU NPIs, works directly with customers to move workloads into production, architects training and inference onboarding frameworks, and collaborates with ML research, performance, and model optimization tooling groups. The posting asks for a bachelor's degree or equivalent, five years of software development, two years leading ML design and infrastructure optimization, and one year with training and inference frameworks. Distinctive: large-scale reference model work and TPU-specific focus.