← All roles

Senior GPU Performance Software Engineer, AI/ML

Google· Engineering
Apply on company site
London, UKFull-timePosted 2w ago

About the role

Senior GPU-focused software engineering role on the Machine Learning (ML)-Omega team to optimize, model, and evaluate GPU systems for Google's internal ML workloads, with emphasis on performance benchmarking and guidance for cloud hardware decisions.

Responsibilities

  • Identify and maintain Large Language Model (LLM) training and serving benchmarks representative of Google production, industry, and ML community; drive XLA GPU/Triton performance and guide XLA releases.
  • Engage with Google product teams like DeepMind to solve ML model performance problems, onboarding new LLM models and products on GPU hardware and enabling large-scale training and serving.
  • Run architecture-level simulations on GPU designs and perform roofline analysis to guide internal teams.
  • Analyze performance and efficiency metrics to identify bottlenecks, and design/implement solutions at Google fleet-wide scale.
  • Run performance benchmarks on GPU hardware using internal and external tools.

Qualifications

  • Bachelor’s degree or equivalent practical experience.
  • 5 years of experience with software development in one or more programming languages.
  • 5 years of experience with data structures and algorithms.
  • 5 years of experience with machine learning algorithms and tools, artificial intelligence, deep learning, or natural language processing.
  • 3 years of experience testing, maintaining, or launching software products, and 1 year of experience with software design and architecture.

Skills

Required:Software developmentData structures and algorithmsMachine learning / AI / DL / NLPSoftware design and architecture

Benefits

  • Google benefits as applicable to role not explicitly listed in description.
Last updated Oct 23, 2026. Interested? Apply on the company site.

About Google

Google Search homepage

llmtext-to-speechvoice-ai-stack
Company profileWebsite

More open roles at Google