ML Systems Engineer: TPU Training & Inference Optimizer

GOOGLE ASIA PACIFIC PTE. LTD.; Singapore; On-site; SGD 180,000 - 260,000; Full time

Job summary

GOOGLE ASIA PACIFIC PTE. LTD. is seeking software engineers to join the AI and Infrastructure team in Singapore.

You will contribute to next-generation technologies that power billions of users, including on TPUs and Cloud AI platforms, from onboarding ML models to optimizing large-scale training and inference workloads. The role emphasizes versatility, leadership, and the ability to work across full-stack systems—hardware, firmware, and software—on scalable cloud infrastructure with high

Qualifications

• Bachelor's degree or equivalent practical experience.
• 1 year of experience leading ML design and optimizing ML infrastructure.
• 1 year of experience with training and inference techniques.

Skills

ML software engineering

System design

Leadership in ML

Education

Bachelor's degree

Master's degree

PhD

Tools

Megatron-LM

DeepSpeed

TensorRT-LLM

vLLM

SGLang

Job description

GOOGLE ASIA PACIFIC PTE. LTD. is seeking software engineers to join the AI and Infrastructure team in Singapore.

You will contribute to next-generation technologies that power billions of users, including on TPUs and Cloud AI platforms, from onboarding ML models to optimizing large-scale training and inference workloads. The role emphasizes versatility, leadership, and the ability to work across full-stack systems—hardware, firmware, and software—on scalable cloud infrastructure with high