Senior Staff Engineer, Cloud Machine Learning Compute Systems

Google Cloud is seeking a Senior Staff Engineer to lead their Machine Learning Compute Systems initiatives. This role combines deep technical expertise in machine learning optimization with system-level engineering. You'll be working on critical projects that impact how billions of users interact with Google's technologies, focusing on optimizing ML models for performance, efficiency, and accuracy. The position offers the opportunity to work with cutting-edge technology in cloud computing and machine learning, while solving complex technical challenges at massive scale.

As a Senior Staff Engineer, you'll be responsible for developing and implementing sophisticated optimization strategies for machine learning models, working with custom kernels, and leveraging deep learning frameworks like JAX. The role requires expertise in both the theoretical and practical aspects of machine learning systems, including model architecture design, hyperparameter optimization, and hardware-level optimizations.

Google Cloud serves customers in over 200 countries, providing enterprise-grade solutions that leverage Google's cutting-edge technology. This position offers competitive compensation ($248,000-$349,000 + bonus + equity + benefits) and the opportunity to work with world-class engineers and researchers. The ideal candidate will combine technical excellence with leadership qualities, bringing fresh ideas to areas including distributed computing, system design, and artificial intelligence.

Working at Google means being part of a company that values diversity, inclusion, and innovation. You'll have the chance to impact billions of users while working with the latest technologies in machine learning and cloud computing. The role offers significant growth opportunities and the ability to influence the direction of Google's machine learning infrastructure.