ML Performance Engineer

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 hybrid role, you will report to a Software Engineering Manager.

You will:

• Collect application/ML model performance traces and analyze for performance optimization opportunities
• Land the optimizations to the application/ML model codebase by evaluating the correctness of the change (including retraining the ML model to evaluate accuracy)
• Ensure the change is net-positive for the current onboard performance
• Add proper testing to ensure optimized performance is tracked and not regressed in the future
• Generalize the optimization at the infra level (compiler, firmware, runtime, framework) and project the potential impact to all applications (ML models) across board
• Set the performance expectation through solid methodology, ensure the Infrastructure team is aligned, and evaluate the performance impact

You have:

• Bachelors in Computer Science or Electrical Engineering
• Proficient in C++ and Python
• 3+ years of experience with application performance optimization
• Proficiency in collaboration with teams in US

We prefer:

• ML performance and compiler
• Computer architecture and performance analysis methodologies
• Robotic application development/optimizations

#LI-Hybrid