Senior Machine Learning Engineer – Simulation Scenario Generation
Zoox
full-time
Posted on:
Location Type: Hybrid
Location: Foster City • California • 🇺🇸 United States
Visit company websiteSalary
💰 $233,000 - $290,000 per year
Job Level
Senior
Tech Stack
NumpyPythonPyTorch
About the role
- Contribute to tooling for AI-based scenario understanding and validation.
- Synthesize realistic AV simulation scenarios with dynamic (e.g., traffic) and static features.
- Integrate and validate LLMs/VLMs and implement other models for complex scenario generation workflows, leveraging techniques like agentic tool use.
- 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.
Requirements
- MS or PhD in Computer Science, Machine Learning, or related field
- 5+ years of industry experience in Machine Learning
- Proficiency in Python and ML libraries (PyTorch, JAX, NumPy, etc.) demonstrated through professional or research projects
- Demonstrated experience in transformer and diffusion architectures
- Practical experience in dataset creation for fine-tuning, system integration of ML models into production, or optimization techniques for low-latency inference systems
Benefits
- 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
- life insurance
Applicant Tracking System Keywords
Tip: use these terms in your resume and cover letter to boost ATS matches.
Hard skills
PythonMachine LearningPyTorchJAXNumPytransformer architecturesdiffusion architecturesdataset creationsystem integrationoptimization techniques
Soft skills
collaborationcommunication
Certifications
MS in Computer SciencePhD in Computer Science