Senior Customer Engineer, AI Infrastructure, Google Cloud

Location: Singapore

Experience owning outcomes and decision making, solving ambiguous problems and influencing stakeholders; deep expertise in domain.

• Bachelor's degree or equivalent practical experience.
• 10 years of experience in developing and deploying models using deep learning frameworks (e.g., TensorFlow, PyTorch, or JAX) specifically on TPU hardware.
• Experience in networking principles, including concepts like collective communication, inter-chip interconnects, and their impact on distributed AI training.

• Experience with lower-level performance tools and techniques (e.g., custom kernel development, XLA compiler familiarity) relevant to optimizing code for Google's TPU chips.
• Experience with leveraging AI hardware and software stacks and platforms to bring up and deploy AI com...