Senior/Staff Machine Learning Engineer (AI)

TL;DR

Senior/Staff Machine Learning Engineer (AI): Developing multimodal large language models to enhance robotaxis' understanding of complex urban environments with an accent on model architecture, training techniques, and integration into the vehicle's decision-making pipeline. Focus on end-to-end ML solutions from research to production, using real-world driving scenarios to improve safety and reliability.

Location: Foster City, CA

Salary: $242,000 - $333,000 a year

Company

Zoox is leveraging the latest in AI to craft algorithms that understand the world and enable safe and efficient navigation in complex environments.

What you will do

• Lead the development of multimodal large language models that enhance understanding of complex urban environments.
• Design effective model architectures and sophisticated training techniques, leveraging sensor data and large-scale data.
• Drive end-to-end ML solutions from research to production, utilizing data pipelines and infrastructure.
• Collaborate with perception, planning, safety, and systems teams to integrate models into the vehicle's decision-making pipeline.
• Validate and optimize solutions using real-world driving scenarios, contributing to the safety and reliability of the autonomous system.

Requirements

• MS or PhD in Computer Science, Machine Learning, or related technical field
• Demonstrated experience training and deploying large language models (LLMs).
• Experience building and maintaining ML training pipelines, including data preprocessing, model training, and evaluation.
• Proficiency in Python and ML libraries (PyTorch, NumPy) demonstrated through professional or research projects.
• Experience training with large-scale datasets.

Nice to have

• Publications in top-tier conferences (CVPR, ICCV, RSS, ICRA)
• Experience with autonomous robotics systems