Senior Machine Learning Engineer - Perception Labeling

Zoox; Foster City (CA); On-site; USD 197,000 - 252,000; Full time

Benefits offered by this job

Health insurance

Stock Appreciation Rights

RSUs

Paid time off

Job summary

Zoox in Foster City, CA seeks an AI engineer to design autolabeling platforms for the Perception stack. You will collaborate with ML engineers, data scientists, and software engineers to build scalable pipelines that auto-label PBs of multimodal data and manage versioned datasets for prototyping and training online models.

You will stay current with autolabeling and unsupervised learning trends, integrate datasets into the Dataloop platform, and contribute to a visionary product roadmap that

Qualifications

• Bachelor's or Master's in CS, engineering, or related field.
• 5+ years of experience in software engineering and data-driven ML development.
• Experience with data management tools and frameworks.
• Excellent communication and cross-functional collaboration skills.

Responsibilities

• Collaborate with ML engineers to research or improve autolabeling algorithms for model training.
• Evaluate autolabel data quality and provide feedback to improve model performance.
• Integrate autolabel datasets and pipelines into the Dataloop platform to enable end-to-end ML workflows.
• Define problems and opportunities to iterate the product roadmap with stakeholders.
• Stay up-to-date with autonomous driving perception and ML trends, especially autolabeling and unsupervised learning.

Skills

Software engineering

ML development

Data management

Communication

Education

Bachelor's or Master's in CS/Engineering

Tools

Ray

Job description

We are seeking an AI engineer to design and develop auto-labeling algorithms and platforms to facilitate AI model development for the Perception component of our autonomous driving stack. In this role, you will work closely with Machine Learning engineers, data scientists, and software engineers to build scalable and high-performance pipelines that auto-label PBs of multimodal data and manage versioned datasets used by ML engineers to prototype and train online models with maximized productivity.

The Perception Labeling and Tools team aims to automate the dataloop and lead data-driven model development platforms and processes. The team is responsible for:

• Data mining tools and data pipelines, such as active learning framework, miner scheduling and visualization tools, CLIP embedding indexes, etc.
• Labeling platforms such as human labeling web apps, labeling task and labeler management platform, auto-labeling pipelines.
• Dataset management app for label data curation and versioned dataset management and model-loop integration.

In this role, you will:

• Collaborate with ML engineers to research, implement, or improve autolabeling algorithms to generate auto-labeled data for model training and prototyping.
• Work with ML engineers and data scientists to evaluate autolabel data quality and seek a feedback loop to continuously improve model performance.
• Work with other software engineers in the team to integrate autolabel datasets and pipelines into the Dataloop platform to deliver impact on end-to-end data-driven ML development.
• Continuously define problems and opportunities to iterate the application with a visionary product roadmap that keeps stakeholders on the same page.
• Stay up-to-date with the latest trends and technologies in autonomous driving perception and machine learning, especially autolabeling and unsupervised learning.

Qualifications

• Bachelor's or Master's degree in Computer Science, Engineering, or a related field.
• 5+ years of experience in software engineering and data-driven ML development.
• Experience with data management tools and frameworks.
• Excellent communication and collaboration skills, with the ability to work effectively in a cross-functional team environment.

Bonus Qualifications

• Experience in scalable AI computation frameworks such as Ray.
• Experience with human labeling or autolabeling.
• Experience in ML-related development in the Autonomous Driving industry.

Compensation and Benefits

There are three major components to compensation for this position: salary, Amazon Restricted Stock Units (RSUs), and Zoox Stock Appreciation Rights. The salary will range from $197,000-$252,000. A sign-on bonus may be part of a compensation package. Compensation will vary based on geographic location, job-related knowledge, skills, and experience.

Zoox also offers a comprehensive package of benefits including paid time off (e.g. sick leave, vacation, bereavement), unpaid time off, Zoox Stock Appreciation Rights, Amazon RSUs, health insurance, long-term care insurance, long-term and short-term disability insurance, and life insurance.