Senior Customer Engineer, AI Infrastructure, Google Cloud

the company will be prioritizing applicants who have a current right to work in Singapore, and do not require the company's sponsorship of a visa.

• 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 the company's TPU chips.
• Experience with leveraging AI hardware and software stacks and platforms to bring up and deploy AI compute clusters.
• Knowledge of AI accelerator hardware (e.g., specific GPU ge...