Staff ML Compiler Performance Engineer – Co-Design Lead

Google is seeking a Staff SWE, ML Compiler Performance Engineer Co-Design in Mountain View, CA. You will tackle pre-silicon ML workloads, drive performance modeling, and contribute to RTL and compiler integration for large-scale AI systems.

You will collaborate across hardware, software, and AI research teams, own critical milestones, and deliver open-source style methodologies to map ML models to TPU performance.