Staff Software Engineer, ML Compilers

Google is seeking a Staff Software Engineer to join their ML Compilers team, focusing on the development and optimization of EdgeTPU compiler systems. This role combines advanced compiler technology with machine learning, requiring expertise in both software engineering and ML systems. The position involves working with cutting-edge AI accelerator technology, optimizing ML workloads, and collaborating with hardware architects to design next-generation systems.

The ideal candidate will have extensive experience in software development, particularly in compiler systems and ML model optimization. They will work on critical projects involving parallelization algorithms, hardware-software interface design, and the efficient mapping of generative AI models to EdgeTPU instructions.

This role offers the opportunity to work at the intersection of compiler technology and machine learning, directly impacting Google's AI hardware capabilities. The position requires collaboration across multiple teams, including hardware architects, ML researchers, and product managers, to drive innovation in AI acceleration technology.

Working at Google, you'll be part of a team that's pushing the boundaries of AI hardware and software integration, with access to world-class resources and the opportunity to impact billions of users. The role combines technical leadership with hands-on development, requiring both deep technical expertise and strong collaborative skills.