Machine Learning Automation Engineer

@ Zoox

Machine Learning Automation Engineer

Posted 1 month ago

This job is still taking applications, but it's been up a while.

About the job

Zoox's SBA team builds ML and data pipelines to help autonomous robots understand failures. Role focuses on designing and deploying agentic LLM triage systems in production while maintaining pipeline health.

Requirements

4+ years in Triage or Performance Analytics
Strong Python experience with CI/CD
Experience with RAG, PyTorch, TensorFlow
Built or managed agentic LLM systems
Managed production pipelines on cloud platforms

Qualifications

PhD or Master's in STEM field
Experience with large datasets
Leadership in AI/ML projects
Strong problem-solving skills

Full 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

Show full description

Similar jobs in Columbus, OH

See all Machine Learning Automation Engineer jobs