2 months ago
Base Salary
$251k - $310k/yr
Responsibilities
- Lead the design and development of multi-sensor, multi-task model architectures for dense scene understanding.
- Architect and develop foundational modeling libraries and software for model development and experimentation.
- Implement sensor encoders, sensor fusion modules, backbones, and temporal aggregation or memory blocks.
- Optimize machine learning models for latency, throughput, memory, and strict compute budgets.
- Build and maintain flexible prototyping pipelines and development environments for rapid model iteration and analysis of training dynamics and learned representations.
- Collaborate with downstream teams on optimization and integration into the Waymo Driver.
Requirements
- Bachelor’s degree in Computer Science or a similar discipline, or an equivalent amount of deep learning experience.
- At least 7 years of experience in machine learning and/or computer vision.
- Strong expertise in Python and C++ with a solid software engineering background.
- Experience with machine learning frameworks such as PyTorch, JAX, or TensorFlow.
Benefits
- Hybrid work arrangement.
- Eligibility for a discretionary annual bonus program, equity incentive plan, and company benefits program, subject to eligibility requirements.
- Full-time position.
Categories
ML EngineeringRobotics
About Waymo
Waymo develops the Waymo Driver, a full-stack autonomous driving system, and operates the Waymo One robotaxi service that offers paid, driverless rides in select U.S. cities. The company monetizes through ride-hailing fares and partnerships integrating its technology on multiple vehicle platforms. Founded in 2009 as Google’s self-driving project, Waymo is headquartered in Mountain View and is an Alphabet subsidiary, with over 10 million rider-only trips and 100+ million autonomous miles on public roads.