Staff ML Accelerator Architect

Waymo's Compute Team is tasked with a critical and exciting mission: We deliver the compute platform responsible for running the fully autonomous vehicle's software stack. To achieve our mission, we architect and create high-performance custom silicon; we develop system-level compute architectures that push the boundaries of performance, power, and latency; and we collaborate closely with many other teammates to ensure we design and optimize hardware and software for maximum performance. We are a multidisciplinary team seeking curious and talented teammates to work on one of the world's highest performance automotive compute platforms.

In this role you'll:

• Analyze workloads and map them efficiently to hardware, proposing novel HW-friendly implementations and projecting performance
• Architect, simulate and design amazing machine learning solutions for our autonomous driving technology
• Work closely with compiler and model developers to influence engineering tradeoffs and future model architectures
• Build scalable tools for simulator modeling and performance evaluation
• Interact with cross-functional engineering teams to identify opportunities and requirements

At a minimum, we'd like you to have:

• BS degree in Computer Science or Computer Engineering or equivalent, or equivalent practical experience
• 8+ years on designing/architecting complex, high performance architectures - CPUs, GPUs and/or ML accelerators - in the industry or through doctoral research
• 3+ years modeling high performance architectures in cycle-aware simulators
• Strong C++ programming and algorithmic problem solving skills

It's preferred if you have:

• Familiarity with ML model architectures and their compute characteristics (bottlenecks, optimization opportunities)
• Track record of analyzing workloads and architecting, delivering novel HW+SW solutions to vastly improve performance, efficiency
• Experience with microarchitecture design (SystemVerilog or HLS)

#LI-Onsite