jobs

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-02T07:54:50+00:00

Google is hiring a mid-level software engineer in Singapore for TPU workload optimization, within the AI and Infrastructure team. The role covers onboarding a stable stack and optimizing training and serving on new TPU NPIs, plus debugging system issues and leading server software design. It asks for a bachelor's degree or equivalent, two years of software development or one with an advanced degree, one year of ML design/infrastructure optimization, and one year with training frameworks such as Megatron-LM or DeepSpeed and inference tools such as TensorRT-LLM or vLLM. On-site, full-time.