Google is seeking a Senior ML Compiler Engineer to join their Silicon team, focusing on developing Machine Learning (ML) compilers for the Tensor TPU to accelerate Generative AI and other complex machine learning models running on custom hardware accelerators. This role combines software engineering expertise with specialized knowledge in compiler development and machine learning optimization.
The position requires deep technical expertise in compiler development, with a focus on mapping ML models to hardware architectures. You'll be working on critical projects that optimize performance, power efficiency, and memory consumption for ML workloads. The role involves close collaboration with both ML researchers and hardware engineers, contributing to the evolution of future accelerators and the improvement of domain-specific compilers.
As part of Google's broader mission to organize the world's information, you'll be working with cutting-edge technology that impacts billions of users. The team combines Google's expertise in AI, Software, and Hardware to create innovative solutions. You'll be particularly involved in productizing ML workloads on the Pixel Edge TPU, including work with Gemini Nano, Camera, Imaging, and Speech ML models.
This is an excellent opportunity for someone with strong compiler experience who wants to work at the intersection of machine learning and hardware optimization. The role offers the chance to impact Google's next-generation AI technologies while working with some of the most advanced ML hardware accelerators in the industry.