Staff Software Engineer, AI/ML Compute

Google is seeking a Staff Software Engineer specializing in AI/ML Compute to join their ML, Systems, & Cloud AI (MSCA) organization. This role is crucial for developing and optimizing machine learning infrastructure that powers Google's services and Google Cloud. The position involves working with cutting-edge AI technologies, including TPUs and Vertex AI platform, while focusing on performance optimization and infrastructure enhancement for AI workloads.

The role requires a blend of deep technical expertise in software development, ML infrastructure, and system architecture, along with leadership capabilities. You'll be working directly with customers and internal teams to benchmark and optimize AI/ML model performance, troubleshoot complex systems, and design specialized ML solutions. The position offers competitive compensation ($197,000-$291,000 base salary) plus additional benefits including bonus and equity.

As part of Google's MSCA organization, you'll be at the forefront of hyperscale computing, working on systems that impact billions of users worldwide through services like Search, YouTube, and Google Cloud. The role provides an opportunity to shape the future of AI infrastructure while working with enterprise customers implementing Gemini models.

This position is perfect for experienced engineers who are passionate about AI/ML systems, have a strong background in software development and architecture, and want to make a significant impact on global-scale infrastructure. The role offers the excitement of working with cutting-edge technology while requiring the ability to collaborate effectively across teams and with customers.