Senior Staff Tech Lead Manager (Simulator Efficiency)

Waymo LLC; San Francisco (CA); Hybrid; USD 200,000 - 260,000; Full time

Benefits offered by this job

Hybrid work model with remote work

On-site meals and snacks

Fitness centers

Commuter benefits

Educational reimbursement

Job summary

Waymo LLC is seeking a technical leader to guide the Simulator Platform across teams, driving architecture improvements and performance optimization for large-scale simulations.

You will shape roadmaps, collaborate with evaluation and infrastructure teams, and push for efficient resource utilization (CPU/RAM/TPU/GPU) across data centers.

Qualifications

• Proven track record setting technical vision and delivering data projects at senior engineering level.
• Excellent communication skills to articulate designs and trade-offs to stakeholders including leadership and partner teams.
• BS/MS in CS, EE, Robotics, Physics, Math, or related field (or equivalent experience).

Responsibilities

• Set technical vision and multi-quarter roadmaps for simulator platform.
• Lead cross-team collaboration to optimize compute usage (CPU/GPU/TPU) and system performance.
• Design and implement architecture improvements for high-throughput simulation at scale.
• Provide guidance across teams to scale simulation and evaluation workloads.

Skills

Technical leadership

C++

Performance optimization

Roadmapping

Cross-team collaboration

Data projects

Education

BS/MS in CS/EE/Robotics/Physics/Math

Tools

CUDA

SIMD

Job description

• The Simulator Team at Waymo builds state-of-the-art simulations of realistic environments for testing and training the Waymo Driver


• We develop industry-leading simulation solutions using advanced ML algorithms that measure and enhance the performance of the Waymo Driver


• The Simulator platform is complex and evolving


• To ensure that Waymo’s Simulation and Evaluation system can scale with new developing Simulation technologies as well as Waymo’s growth, this technical leadership role will look across multiple teams and guide strategy for the entire Simulator software stack


• This can be everything from low-level optimization up through higher level configuration, all with an eye towards improved utilization across various resources (CPU/RAM/TPU/GPU)


• You will lead the Simulator Platform’s efficiency efforts and represent the team in wider Waymo performance-related efforts


• Develop Simulator’s architecture improvements and frameworks that maximize performance and compute utilization


• Evolve compute usage of simulation to enable continued scaling where the system runs efficiently in our data centers


• Collaborate with evaluation and infrastructure teams to identify and improve compute performance bottlenecks across the stack


• Design, build, and optimize the simulator to enable high throughput and high compute utilization for both single instance simulation and large scale simulations


• Developing necessary high scale performance evaluation, debugging and software change management processes


• Optimizing system resource usage to simulation at scale: minimizing CPU utilization and latency, minimizing RAM consumption, intelligently determining which computations should happen on CPU, GPU, and TPU

Benefits

• Medical, dental, and vision insurance for employees and dependents


• Employee assistance programs focused on mental health


• Personalized workplace adjustments for diverse needs and abilities, including physical, mental, and neurodivergent considerations


• Access to mental health apps


• Onsite wellness centers


• Second medical opinion for you and your loved ones


• Counseling services


• Support programs including menopause benefit


• Medical advocacy program for transgender employees


• Competitive compensation


• Regular bonus and equity performance grant opportunities


• Generous 401(k) and regional retirement plans


• Annual cross-company compensation review and pay equity analysis


• 1-on-1 financial coaching


• Fertility and growing family assistance


• Parental leave and baby bonding leave


• Elder care and support


• Survivor income benefit


• Backup childcare


• Caregiver leave


• Paid time off, including vacation, bereavement, sick leave, parental leave, disability, and holidays


• Jury duty leave


• Military leave


• Hybrid work model with remote work opportunities also available


• Educational reimbursement


• Peer learning and coaching platform


• Donation matching programs


• Volunteer hours


• Employee resource groups


• Internal community groups and local culture clubs


• Inspiring spaces to work, recharge, and collaborate with fellow Waymonauts


• On-site meals and snacks


• Fitness centers, massage programs, and ergonomic support


• On-demand fitness, wellbeing, and cooking classes


• Commuter benefits

Setup collaboration structures across team boundariesProven track record of setting technical vision, driving multi-quarter roadmaps, and delivering impactful data projects as a technical lead for senior engineering teams, with demonstrated ability to influence across organizational boundariesExcellent communication skills, with the ability to articulate complex technical designs, trade-offs, and strategies to diverse stakeholders, including senior leadership and partner engineering, data scientists, and product teamsProficient in C++6+ years experience on large scale or high complexity systemDefine roadmap/portfolio of projects optimizing for end results overall across all aspects of the problem spaceExtensive experience acting as technical lead in performance/software infrastructure domainBS/MS in Comp Sci, EE, Robotics, Physics, Math, or related field (or equivalent experience)Solid software management experience in infrastructure/systems/performance domain optimizing end to end system for high performance metricsExperience in roboticsExperience in low level optimization techniques, frameworks (SIMD/CUDA) and ML performance/frameworksExperience in large scale software re-architecture projectsExperience in large scale evaluation techniques/data science and building performance metrics/tooling