Technial Lead Manager, ML Compiler, GPU

Google is a global technology company that provides cutting-edge cloud solutions and tools for digital transformation.
Machine Learning
Staff Software Engineer
In-Person
5,000+ Employees
8+ years of experience
AI · Enterprise SaaS

Description For Technial Lead Manager, ML Compiler, GPU

Google is seeking a Technical Lead Manager for their ML Compiler and GPU team in Zürich, Switzerland. This is a crucial role that combines technical leadership with people management, focusing on building compiler technology that powers Google's machine learning frameworks including JAX, PyTorch/XLA, and TensorFlow.

The position requires a strong technical background with 8 years of software development experience and 3 years of technical leadership. You'll be at the forefront of developing cutting-edge ML workloads like Gemini and next-generation Large Language Models (LLMs), while managing and growing a team in Zürich.

As a Technical Lead Manager, you'll be responsible for enhancing the GPU software stack, optimizing performance for ML workloads, and fostering collaboration between customers and GPU manufacturers. The role involves both technical decision-making and team leadership, requiring expertise in compiler technology, specifically LLVM/MLIR and HPC compilation.

Working at Google offers the opportunity to impact billions of users while working with cutting-edge technology. The company provides a collaborative environment where you'll work with exceptional engineers across multiple teams and locations. You'll be part of Google Cloud, which serves customers in more than 200 countries, helping organizations digitally transform their businesses.

The ideal candidate will have experience with ML frameworks such as TensorFlow, PyTorch, JAX, or Spark ML, combined with strong leadership skills. You'll be expected to balance technical contributions with team management, ensuring both project success and team growth. This role offers the chance to work on some of the most advanced ML infrastructure in the industry while leading a team at one of the world's most innovative companies.

Last updated 6 days ago

Responsibilities For Technial Lead Manager, ML Compiler, GPU

  • Enhance the GPU software stack so that it performs faster and more efficiently on cutting-edge ML workloads
  • Launch new GPU optimizations end-to-end, including performance and correctness testing
  • Foster deep collaboration between customers, partners (e.g. GPU manufacturers) and the GPU compiler team
  • Support and develop members of the team and grow the team overall in Zürich

Requirements For Technial Lead Manager, ML Compiler, GPU

Python
Java
JavaScript
  • Bachelor's degree in Computer Science or equivalent practical experience
  • 8 years of experience with software development in one or more programming languages (e.g., Python, C, C++, Java, JavaScript)
  • 3 years of experience in a technical leadership role
  • 2 years of experience in people management, supervision/team leadership role
  • Experience designing and deploying with ML frameworks such as TensorFlow, PyTorch, JAX, Spark ML (preferred)
  • Experience working with compiler technology, specifically LLVM/MLIR and HPC compilation (preferred)

Interested in this job?

Jobs Related To Google Technial Lead Manager, ML Compiler, GPU

Staff Software Engineer, Machine Learning Borglet Accelerators

Staff Software Engineer position at Google focusing on machine learning infrastructure and hardware acceleration, offering competitive compensation and the opportunity to work on cutting-edge technology.

Staff Software Engineer, AI/ML Natural Language Processing, Google Ads

Lead AI/ML Natural Language Processing initiatives at Google Ads, developing large-scale solutions and providing technical leadership for advertising technology innovations.

Senior Staff Software Engineer, Recommendations, Rankings, Predictions

Lead machine learning and recommendation systems development at Google as a Senior Staff Software Engineer, shaping how billions of users discover content on YouTube.

Senior Staff Software Engineer, Cloud Specialized Generative AI

Lead Google Cloud's Generative AI initiatives as a Senior Staff Software Engineer, developing next-gen technologies and solutions for enterprise customers worldwide.

Staff Software Engineer, Machine Learning, Google Assistant

Lead Machine Learning engineering role at Google, developing AI solutions for Google Assistant with competitive compensation and opportunity to impact billions of users.