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 in 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.