Senior Software Engineer, GPU Performance, Google Scale

Google Cloud is seeking a Senior Software Engineer specializing in GPU Performance to join their ML, Systems, & Cloud AI (MSCA) organization. This role is critical in shaping the future of AI and accelerated computing at Google, focusing on optimizing GPU performance for machine learning workloads at scale.

The position involves working with cutting-edge GPU technologies and ML infrastructure, developing optimizations that impact billions of users worldwide. You'll be working on sophisticated performance models and benchmarks, utilizing the latest generation of GPUs and tools, while collaborating with experts in ML, compiler design, and systems architecture.

The role combines deep technical expertise in GPU architecture and optimization with machine learning infrastructure knowledge. You'll be responsible for improving the entire GPU software stack, from model design to low-level kernel optimization, and working with technologies like OpenXLA, JAX, and Triton.

Google offers a competitive compensation package including base salary, bonus, equity, and comprehensive benefits. The position provides opportunities to work with state-of-the-art technology and influence the direction of GPU computing at Google scale. You'll be part of a team that's essential to Google's machine learning infrastructure and cloud services.

This is an excellent opportunity for experienced engineers passionate about high-performance computing and machine learning, offering the chance to work on challenging problems while making a significant impact on Google's AI infrastructure and products. The role provides exposure to both cutting-edge technology and strategic partnerships with hardware vendors, making it an ideal position for those looking to shape the future of AI computing.