Senior Software Engineer, Machine Learning Performance

London, England, United Kingdom
Jul 16, 2024
Jul 16, 2025
Onsite
Full-Time
5 Years
Job Description

Google's software engineers develop the next-generation technologies that change how billions of users connect, explore, and interact with information and one another. Our products need to handle information at massive scale, extending well beyond web search. We're looking for engineers who bring fresh ideas from all areas, including information retrieval, distributed computing, large-scale system design, networking and data storage, security, artificial intelligence, natural language processing, UI design, and mobile; the list goes on and is growing every day. As a software engineer, you will work on a specific project critical to Google’s needs with opportunities to switch teams and projects as you and our fast-paced business grow and evolve. We need our engineers to be versatile, display leadership qualities, and be enthusiastic to take on new problems across the full-stack as we continue to push technology forward. 

Minimum Qualifications

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

Preferred Qualifications

  • Master's degree or PhD in Computer Science or a related technical field.
  • 1 year of experience in a technical leadership role.
  • Experience with performance analysis and optimization, including system architecture, performance modeling, or similar.
  • Experience with distributed development and large-scale data processing.
  • Experience with compiler optimizations or related fields. 

Responsibilities

  • Identify and maintain Large Language Models (LLMs) training and serving benchmarks that represent Google production standards, industry benchmarks, and Machine Learning community benchmarks. Identify performance opportunities and drive TensorFlow/JAX TPU performance, and gate TF/JAX releases.
  • Engage with Google Product teams to solve LLM performance issues. Onboard new LLM models and products on Google’s new TPU hardware, enabling efficient training on a large scale.
  • Analyze performance and efficiency metrics to identify bottlenecks, design, and implement solutions.
  • Collaborate with tooling and fleet metrics subteams to develop tools for tracking performance and efficiency, and to extract metrics from running Google workloads.
  • Explore techniques for model and data efficiency, new machine learning model architectures, optimizers, and training techniques to address machine learning tasks. Develop new methods to reduce the amount of labeled/unlabeled data needed to train a model.

About Google Cloud

Google Cloud accelerates every organization’s ability to digitally transform its business and industry. We deliver enterprise-grade solutions leveraging Google’s cutting-edge technology and tools that help developers build more sustainably. Customers in over 200 countries and territories trust Google Cloud as their partner to enable growth and solve their most critical business challenges.

Benefits

At Google, we offer a competitive compensation package that includes salary, bonuses, and comprehensive benefits. We also provide opportunities for career growth, professional development, and a collaborative work environment where you can innovate and drive impact.

Equal Opportunity Employer

Google is proud to be an equal opportunity and affirmative action employer. We are committed to building a workforce that is representative of the users we serve, creating a culture of belonging, and providing equal employment opportunities regardless of race, color, religion, gender, sexual orientation, gender identity/expression, national origin, disability, age, genetic information, veteran status, marital status, pregnancy or related condition (including breastfeeding), or any other basis protected by law.

How To Apply

If you’re ready to join Google and make an impact through advanced machine learning performance optimizations, apply now through our careers portal. We look forward to reviewing your application.