Senior Staff Machine Learning Engineer, Perception

We've successfully deployed self-driving cars; now the focus is scaling. This means larger models trained on more data for better generalization. But it's not just about size; we need to automate more and optimize resources.

In this role, you'll shape our tech roadmap by aligning product needs with technology development, balancing current solutions with future innovations. You'll build active learning and ML-aided labeling workflows to tackle rare, "longtail" issues—critical when driving millions of miles daily. As models evolve, so do interfaces and data demands. You'll be key in aligning product vision with these new technology designs.

You'll partner closely with Product and Engineering teams, especially those focused on data and automation. Your responsibilities include supporting freeway expansion, streamlining evaluations, conducting deep dives, improving communication with partners, redefining interfaces for dense perception tasks, and boosting labeling efficiency. You'll also drive our data flywheel and autonomous learning initiatives. We're at the start of an exponential scaling curve, with each new magnitude bringing unique challenges and continuous tech evolution. This role offers significant growth opportunities, including mentoring to foster internal advancement.

You will: 

• Collaborate with Perception leads to align short- and long-term product needs with technology investments.

• Deeply understand technology capabilities and roadmaps to ensure alignment with product requirements and guide product roadmap development.

• Partner with Perception leads to define problems, design interfaces, and establish metrics and evaluation methodologies.

• Work with Perception and Engineering teams to develop active learning loops and model-aided human labeling workflows, optimizing efficiency and value.

• Drive the automation of Perception model evaluation, verification, and production monitoring with Engineering partners.

• Accelerate collaboration with Product and Engineering teams to solve complex product problems leveraging deep technical knowledge.

You have:

• Master's degree or PhD in Computer Science, Engineering, or a related technical field.

• 6+ years of experience developing and deploying ML-driven CV/Perception systems, and 3+ years of experience working on Perception systems for ADVs.

• 3+ years of experience in a technical leadership role, collaborating with product teams and setting technical directions in large ML Engineering organizations.

• 3+ years of experience developing and taking ML systems to production, and monitoring and maintaining them in production.

• Proficiency in C++, Python, and modern deep learning toolkits like PyTorch or JAX.

• Strong communication skills, a passion to make a business impact, and a drive for engineering excellence.

We prefer: 

• Experience developing ML-aided human labeling workflows.

• Experience developing and productizing large-scale vision, video, or multi-modal foundation models.

• Familiarity with end-to-end models and their development challenges.

• Agility in a fast-paced environment.

In this hybrid role, you will report to the Engineering Director in Perception