This role involves building compiler technology for machine learning frameworks at Google, including JAX, PyTorch/XLA, and TensorFlow.  Responsibilities include enhancing the GPU software stack for performance and efficiency on cutting-edge ML workloads, launching new GPU optimizations, fostering collaboration with customers and partners, and developing team members in Zurich. The position requires managing engineers across multiple teams and locations, overseeing large-scale projects, and contributing to product strategy.  The ideal candidate will have extensive experience in software development (Python, C, C++, Java, JavaScript), compiler technology (LLVM/MLIR, HPC compilation), and ML frameworks (TensorFlow, PyTorch, JAX, Spark ML), along with proven technical leadership and people management skills.

<p>Check out our career opportunities at goo.gle/3DLEokh</p>

Google

Senior Software Engineering Manager, Infrastructure, Google Cloud Data Management

Web Solutions Engineer, University Graduate, 2025

Security Sales Specialist, Google Public Sector

Data Center Operations Manager, Server Operations (English, Dutch)

Staff Software Engineer, Site Reliability Engineering, Google Cloud

Early Career Software Engineer, Black Community Inclusion

Account Manager, People with Disabilities (English, Portuguese)

Software Engineer, Black Community Inclusion

Software Engineer III, Site Reliability Engineering, Google Cloud

Software Engineer, People with Disabilities

<div jscontroller="iYVHDc" class="DkhPwc" jsname="tIk9qd" data-id="100573241981969094" jsaction="A6AxKc:CXOOvf;"><div class="KwJkGe"><h3>Minimum qualifications:</h3><ul>
<li>Bachelor's degree in Computer Science or equivalent practical experience.</li>
<li>8 years of experience with software development in one or more programming languages (e.g., Python, C, C++, Java, JavaScript).</li>
<li>3 years of experience in a technical leadership role; overseeing projects, with 2 years of experience in a people management, supervision/team leadership role.</li>
</ul><br><h3>Preferred qualifications:</h3><ul>
<li>Experience designing and deploying with one or more from the following ML frameworks such as TensorFlow, PyTorch, JAX, Spark ML.<br></li>
<li>Experience working with compiler technology, specifically LLVM/MLIR and HPC compilation.</li>
</ul></div><div class="aG5W3"><h3>About the job</h3><p>Like Google's own ambitions, the work of a Software Engineer goes beyond just Search. Software Engineering Managers have not only the technical expertise to take on and provide technical leadership to major projects, but also manage a team of Engineers. You not only optimize your own code but make sure Engineers are able to optimize theirs. As a Software Engineering Manager you manage your project goals, contribute to product strategy and help develop your team. Teams work all across the company, in areas such as information retrieval, artificial intelligence, natural language processing, distributed computing, large-scale system design, networking, security, data compression, user interface design; the list goes on and is growing every day. Operating with scale and speed, our exceptional software engineers are just getting started -- and as a manager, you guide the way.<br><br>With technical and leadership expertise, you manage engineers across multiple teams and locations, a large product budget and oversee the deployment of large-scale projects across multiple sites internationally.<br></p>
<p>In this role, you will build the compiler technology that powers machine learning frameworks at Google, including JAX, PyTorch/XLA and TensorFlow, and cutting-edge ML workloads like Gemini and next-generation Large-Language Models (LLMs). You will also build the MLIR compiler infrastructure and owns the XLA GPU compiler.</p><p>Google Cloud accelerates every organization’s ability to digitally transform its business and industry. We deliver enterprise-grade solutions that leverage Google’s cutting-edge technology, and tools that help developers build more sustainably. Customers in more than 200 countries and territories turn to Google Cloud as their trusted partner to enable growth and solve their most critical business problems.</p></div><div class="BDNOWe"><h3>Responsibilities</h3><ul>
<li>Enhance the GPU software stack so that it performs faster and more efficiently on cutting-edge ML workloads.<br></li>
<li>Launch new GPU optimizations end-to-end, including performance and correctness testing.<br></li>
<li>Foster deep collaboration between customers, partners (e.g. GPU manufacturers) and the GPU compiler team.<br></li>
<li>Support and develop members of the team and grow the team overall in Zürich.<br></li>
</ul></div></div>

Technial Lead Manager, ML Compiler, GPU