Zoox

Machine Learning Engineer, Simulation Scenario Generation

Zoox

full-time

Posted on:

Location Type: Hybrid

Location: Foster City • California • 🇺🇸 United States

Visit company website
AI Apply
Apply

Salary

💰 $151,000 - $257,000 per year

Job Level

JuniorMid-Level

Tech Stack

NumpyPythonPyTorch

About the role

  • Integrate and validate LLMs/VLMs and implement other models for complex scenario generation workflows, leveraging techniques like advanced prompting, agentic tool use, and more.
  • Contribute to tooling for AI-based scenario understanding and validation.
  • Collect data and design metrics to drive business intelligence, product iteration, and model fine-tuning.
  • Collaborate directly with internal customers and partner teams to provide generative AI solutions for their test creation workflows.
  • Directly contribute to the safety and reliability of Zoox's autonomous software.
  • Integrate, implement, and optimize next-generation AV scenario generation workflow; extend AI assistant toward full scenario creation automation from natural language test specification.

Requirements

  • MS or PhD in Computer Science, Machine Learning, or related field
  • 2+ years of industry experience in Machine Learning
  • Solid understanding of LLM or NLP concepts
  • Proficiency in Python and ML libraries (PyTorch, NumPy) demonstrated through professional or research projects
  • Practical experience in dataset creation for fine-tuning, system integration of ML models into production, or optimization techniques for low-latency inference systems (bonus)
  • Familiarity with autonomous vehicles, robotics, and/or complex simulation environments (bonus)
  • Hands-on experience in areas like program synthesis, diffusion models, and/or formal methods/V&V (bonus)
  • Relevant publications in conferences (e.g., CVPR, ICCV, RSS, and/or ICRA) (bonus)
Benefits
  • paid time off (e.g. sick leave, vacation, bereavement)
  • unpaid time off
  • Zoox Stock Appreciation Rights
  • Amazon Restricted Stock Units (RSUs)
  • Health insurance
  • Long-term care insurance
  • Long-term and short-term disability insurance
  • Life insurance
  • Sign-on bonus may be offered

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard skills
Machine LearningNatural Language ProcessingPythonPyTorchNumPydataset creationsystem integrationoptimization techniquesprogram synthesisdiffusion models
Soft skills
collaborationcommunicationproblem-solvingdata analysisbusiness intelligence
Certifications
MS in Computer SciencePhD in Computer Science
PayPal

Senior Machine Learning Technical Lead

PayPal
Seniorfull-time$188k–$324k / yearCalifornia, Illinois, Texas · 🇺🇸 United States
Posted: 3 hours agoSource: paypal.wd1.myworkdayjobs.com
Google Cloud PlatformHadoopPythonPyTorchSQLTensorflow
NVIDIA

Senior Product Manager – AI Infrastructure

NVIDIA
Seniorfull-time$208k–$380k / yearCalifornia · 🇺🇸 United States
Posted: 3 hours agoSource: nvidia.wd5.myworkdayjobs.com
CloudKubernetes
General Motors

Staff Software Engineer – AI Platform

General Motors
Leadfull-timeCalifornia · 🇺🇸 United States
Posted: 1 day agoSource: generalmotors.wd5.myworkdayjobs.com
Greenlight

AI/ML Engineer

Greenlight
Mid · Seniorfull-time$122k–$180k / yearCalifornia, Colorado, New York, Washington · 🇺🇸 United States
Posted: 2 days agoSource: jobs.lever.co
AWSFlaskJavaJavaScriptJenkinsKotlinMicroservicesNode.jsPrometheusPythonScala