Staff Software Engineer, Machine Learning Runtime, Google Cloud

Google is seeking a Staff Software Engineer to join their Machine Learning Runtime team in Google Cloud. This role is part of the Core team, which builds the technical foundation behind Google's flagship products. As a Staff Software Engineer, you'll be at the forefront of developing and optimizing machine learning infrastructure that powers Google's diverse ML ecosystem.

The position offers a compelling opportunity to work on critical projects that impact Google's technical direction across the company. You'll be responsible for leading the design and implementation of specialized ML solutions, managing framework migrations, and optimizing ML infrastructure. The role requires deep expertise in machine learning technologies, particularly in areas such as TensorFlow, JAX, and PyTorch runtimes.

Working at Google means joining a team that builds essential building blocks for excellent, safe, and coherent user experiences while driving innovation for every developer. The Core team has a unique mandate to impact important technical decisions across the company, breaking down technical barriers and strengthening existing systems.

The compensation package is highly competitive, with a base salary range of $189,000-$284,000, plus additional benefits including bonus and equity opportunities. Google offers a comprehensive benefits package and is committed to building a diverse, inclusive workplace where all employees can thrive.

This role is perfect for experienced engineers who are passionate about machine learning infrastructure and want to make a significant impact at scale. You'll work with cutting-edge technologies, lead cross-functional teams, and help shape the future of Google's ML ecosystem. The position requires strong technical leadership skills, extensive experience in ML infrastructure, and the ability to drive complex projects from conception to completion.

If you're excited about solving complex technical challenges, leading high-impact projects, and working with a team that influences Google's technical direction, this role offers an exceptional opportunity to advance your career while making a meaningful contribution to Google's ML infrastructure.