Machine Learning Automation Engineer

Job Description

Job Description

Within the System Behavior Analysis (SBA) organization, the Behavioral Automated Triage (BAT) team builds the ML, deep learning, and data pipelines that help Zoox understand and triage failures our robots encounter on the road. You will build agentic, LLM-powered triage systems and run them in production, keeping pipelines healthy that teams across Zoox use daily, while working closely with stakeholders across safety, planning, and quality assurance.

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

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.