Senior ML Systems Engineer: TPU/GPUs & Cloud

Google; Greater London; On-site; GBP 120,000 - 180,000; Full time

Job summary

Google Cloud is hiring software engineers to advance AI framework development and ML infrastructure. You will enable PyTorch models on TPUs/GPUs, optimize performance, and collaborate with teams across Google to push ML capabilities in Cloud environments.

The role focuses on large-scale ML systems, compiler support, and open-source contributions, with opportunities to work on cutting-edge AI models and tooling.

Qualifications

• Bachelor's degree in Computer Science, related technical field, or equivalent practical experience.
• 5 years of experience with software development in one or more programming languages (e.g., C++, C, or Python).
• Experience in leading cross-functional software oriented projects from inception to delivery.
• 5 years of experience with data structures and algorithms.
• Experience with ML frameworks (e.g., TensorFlow, JAX, PyTorch) and hardware performance analysis, particularly with TPUs and GPUs.
• Experience with Compilers and Compiler construction.
• Experience programming in Python.
• Ability to effectively navigate and communicate in a fast-paced environment.

Responsibilities

• Work on AI framework development to enable PyTorch models on Google Cloud's TPUs/GPUs and tune for peak performance.
• Provide comprehensive support for ML frameworks and compilers on Cloud TPUs and GPUs, enabling training and deployment of the most advanced ML models.
• Enable PyTorch models at massive scale for generative models, computer vision, machine translation, language modeling, rankings and recommendations, speech recognition, etc.
• Collaborate with other Google teams and leading researchers to continuously bring ML capabilities to our PyTorch-in-Cloud offering.
• Contribute to open-source software development.

Skills

Python

Team collaboration

Communication in fast-paced env

Education

Bachelor's degree in CS or related field

Tools

TensorFlow

JAX

PyTorch

TPU

GPU

Compiler construction

Job description

Google Cloud is hiring software engineers to advance AI framework development and ML infrastructure. You will enable PyTorch models on TPUs/GPUs, optimize performance, and collaborate with teams across Google to push ML capabilities in Cloud environments.

The role focuses on large-scale ML systems, compiler support, and open-source contributions, with opportunities to work on cutting-edge AI models and tooling.