Staff Software Engineer, Cloud ML Compute Services

Google Cloud is seeking a Staff Software Engineer to join their Cloud ML Compute Services team, focusing on building and supporting Google Cloud Platform's Cloud TPU and GPU services. This role is perfect for experienced engineers passionate about machine learning infrastructure and high-performance computing.

The position offers an exciting opportunity to work at the intersection of cloud computing and machine learning, developing next-generation technologies that impact billions of users. As part of the Cloud ML Compute Services team, you'll be responsible for the ML frameworks, tools, models, and processes that achieve scale and performance in ML workloads in Google Cloud.

The ideal candidate brings 8+ years of software development experience, with deep expertise in machine learning algorithms and infrastructure. You'll work with cutting-edge technologies like PyTorch and JAX, optimizing LLM training and inference performance on TPUs. The role involves collaboration with various teams and direct interaction with Cloud TPU power users to solve complex technical challenges.

This is a unique opportunity to join Google Cloud, a trusted partner for organizations worldwide, offering competitive compensation ($189,000-$284,000 + bonus + equity + benefits) and the chance to work on infrastructure that powers the future of machine learning. You'll be part of a team that's pushing the boundaries of AI technology while building solutions that help companies operate more efficiently and adapt to changing needs.

The role combines technical leadership with hands-on development, requiring both deep technical expertise and the ability to collaborate across teams. You'll be at the forefront of AI infrastructure development, working with the latest models, tools, and techniques while contributing to Google Cloud's mission of accelerating digital transformation across industries.