Senior Machine Learning Engineer, Zoox

• MS or PhD in Computer Science or related field
• Background in deep learning solutions for VLM and VLA models
• Track record in post-training large-scale models, CPT, SFT, RL
• Hands-on experience with production ML pipelines, including dataset creation, training frameworks, and metrics
• Expertise in Python libraries (PyTorch, NumPy, Pandas, VLLM)
• You do not need to match every listed expectation to apply for this position

Desirable

• As an engineer in the Scene Understanding team, you will develop advanced Vision-Language-Action (VLA) models that perceive our vehicle's surroundings to identify hazards and make driving suggestions
• You will utilize VLA models for detecting rare events and ensuring safe driving in these situations
• You'll work with state-of-the-art machine learning models that operate in real-time on our robotaxi platform with minimal latency
• Collaborating with world-class engineers and researchers across sensors, planning, and other teams, you'll have access to premium sensor data and cutting-edge infrastructure to validate your algorithms in real-world conditions
• Design and train Vision-Language-Action (VLA) solutions for robotaxis
• Lead end-to-end data strategy, including mining, auto-labeling, and dataset construction to power our ML flywheel
• Lead the full post-training stack for VLMs and VLAs, including Continual Pre-training (CPT) on domain-specific driving data, Supervised Fine-Tuning (SFT) for instruction following
• Utilize our large-scale data pipelines and ML infrastructure to research, prototype, and deploy solutions that improve driving behavior
• Partner with cross-functional teams to integrate perception signals