Senior AI Infrastructure Engineer - TPU & Cloud

Google India; Singapore; On-site; SGD 90,000 - 120,000; Full time

Job summary

Google India is seeking a Customer Engineer in Singapore to lead customer interactions on AI technology, leveraging Google Cloud's capabilities. This role demands experience in deep learning, networking principles, and performance optimization, particularly on TPU hardware.

As a key support to Google Cloud Sales, you will design and implement AI solutions, profiling customer models and advising on best practices. The position offers an opportunity to work with diverse clients, enhancing their operations with Google Cloud's advanced AI infrastructure.

Qualifications

• 10 years of experience in developing and deploying models using deep learning frameworks specifically on TPU hardware.
• Experience in networking principles relevant to distributed AI training.
• Knowledge of AI accelerator hardware and the AI infrastructure market.

Responsibilities

• Design and implement complex multi-host AI solutions on Google Cloud TPUs.
• Conduct performance profiling and optimization for TPU architecture.
• Advise on integrating ML workflows with the Google Cloud AI platform.

Skills

Deep learning frameworks (TensorFlow, PyTorch, JAX)

Networking principles

Custom kernel development

AI hardware and software stacks

AI infrastructure market knowledge

Education

Bachelor's degree or equivalent practical experience

Job description

Google India is seeking a Customer Engineer in Singapore to lead customer interactions on AI technology, leveraging Google Cloud's capabilities. This role demands experience in deep learning, networking principles, and performance optimization, particularly on TPU hardware.

As a key support to Google Cloud Sales, you will design and implement AI solutions, profiling customer models and advising on best practices. The position offers an opportunity to work with diverse clients, enhancing their operations with Google Cloud's advanced AI infrastructure.