Staff ML Engineer, Perception Research

Highlights

• Train and deploy multimodal LLMs and world models for 3D perception.
• Build perception using camera, LiDAR, and radar sensor data at Waymo scale.
• Advance autonomous driving toward safe operation in dozens of cities.

What they offer

• Health, dental, vision, life and disability insurance
• 401(k) with company match
• 20 days vacation, 13 holidays, sick time and parental leave
• Discretionary annual bonus program and equity incentive plan

Ideal candidate

Research-minded ML engineer with deep foundation-model experience who thrives on hard perception problems and shipping real-world autonomy. Collaborative and experimental, eager to push multimodal learning into safe self-driving.

Requirements

• PhD or Masters in CS, ML, Robotics or similar with 4+ years industry or post-doc research experience
• Proficiency implementing scalable distributed model training flows such as data parallel, FSDP and sharding
• Proficiency in JAX, Flax, and potentially TensorFlow/PyTorch
• Willingness to work with complexity of globally distributed inference infrastructure
• Hands-on experience optimizing training and inference of Transformer architectures