Machine Learning Automation Engineer

Jobzhr; Foster City, Northern (CA, KY); Hybrid; USD 180,000 - 225,000; Full time

Job summary

Zoox in Foster City, CA seeks a highly skilled ML/DL engineer to design and improve algorithms on large-scale data to automate test and triage workflows.

You will build and run agentic LLM systems (Claude or Gemini) that automate triage from prototype to production, monitor multiple production pipelines, respond to issues, and fix root causes while collaborating with data science, ML, autonomous drive planning, and QA teams.

Qualifications

• PhD or Master's in a STEM field with 4+ years in triage, performance analytics, or ML across enterprise contexts.
• Strong productionization and CI/CD experience in Python; dashboards, evaluation, and data manipulation.
• Experience with RAG, PyTorch, TensorFlow, or HuggingFace; agentic LLM systems in production.
• Managed multiple production pipelines on Databricks and AWS; resolved production issues.

Responsibilities

• Design and improve ML/DL algorithms on large-scale data to automate test and triage workflows.
• Build and run agentic LLM systems that automate triage, from prototype to production.
• Run multiple production pipelines: monitor them, respond to issues, and fix the root causes.
• Work with stakeholders across data science, ML, autonomous drive planning, and quality assurance.
• Build monitoring tools and use them to keep improving your algorithms in production.

Skills

PhD/Master in STEM field

Python

CI/CD

Dashboards

RAG

PyTorch

TensorFlow

HuggingFace

Agentic LLMs

Databricks

AWS

Education

PhD or Master's in STEM field

Tools

Databricks

AWS

Looker

Dash

Job description

In This Role, You Will

Design and improve ML/DL algorithms on large-scale data to automate test and triage workflows.

Build and run agentic LLM systems (e.g., Claude or Gemini) that automate triage, from prototype to production.

Run multiple production pipelines: monitor them, respond to issues, and fix the root causes.

Work with stakeholders across data science, ML, autonomous drive planning, and quality assurance.

Build monitoring tools and use them to keep improving your algorithms in production.

Qualifications

• PhD or Master's in a STEM field and 4+ years in Triage, Performance Analytics, for fortune 500 including infrastructure as code, ML, NLP, or deep learning.
• Strong productionization, CI/CD experience for medium to large projects Python,, with a solid grasp of dashboards, eval, algorithms and data manipulation.
• Experience with RAG, PyTorch, TensorFlow, or HuggingFace
• Has built or managed agentic LLM systems (Claude or Gemini) in a professional setting with large dataset to satisfy multiple stakeholders.
• Has run multiple production pipelines on platforms like Databricks and AWS, including fixing production issues.

Bonus Qualifications

• Experience with self correcting models, installation and maintenance of LLMs, data visualization tools like Looker/Dash/Databricks Dashboards
• Experience in Performance Monitoring at Scale, Support environments, autonomous driving, robotics.
• Strong verbal and written communication skills

$180,000 - $225,000 a year

Base Salary Range

There are three major components to compensation for this position: salary, Amazon Restricted Stock Units (RSUs), and Zoox Stock Appreciation Rights. A sign-on bonus may be offered as part of the compensation package. The listed range applies only to the base salary. Compensation will vary based on geographic location and level. Leveling, as well as positioning within a level, is determined by a range of factors, including, but not limited to, a candidate\'s relevant years of experience, domain knowledge, and interview performance. The salary range listed in this posting is representative of the range of levels Zoox is considering for this position.

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.

We may use artificial intelligence (AI) tools to support parts of the hiring process, such as reviewing applications, analyzing resumes, or assessing responses and identifying potential inconsistencies or verification signals in application materials based on available information. These tools assist our recruitment team but do not replace human judgment. Final hiring decisions are ultimately made by humans. If you would like more information about how your data is processed, please contact us.