Zoox

Company Name: Zoox
Job Title: Senior/Staff Software Engineer - Prediction Machine Learning Engineer
Job Description: Job Title: Senior/Staff Software Engineer – Prediction Machine Learning Engineer Role Summary Lead the design, development, and deployment of machine learning models that predict future behavior of agents in autonomous driving scenarios. Work cross‑functionally with perception, planning, and simulation teams to integrate prediction into vehicle behavior and validation pipelines. Expectations - Deliver scalable, production‑ready ML systems that support on‑vehicle and off‑vehicle use cases. - Drive innovation in behavior forecasting and motion planning through advanced algorithm research. - Provide measurable performance metrics and analytical tools for continuous improvement. - Communicate findings and recommendations clearly to engineering and research stakeholders. Key Responsibilities - Develop and refine transformer‑based models for multi‑agent behavior prediction. - Build end‑to‑end ML pipelines: data collection, preprocessing, training, validation, and deployment. - Implement metrics pipelines and diagnostic tools to assess model accuracy, latency, and robustness. - Collaborate with perception, planning, and simulation teams to align model outputs with system requirements. - Conduct experiments, analyze results, and iterate on model architecture and training strategies. - Document algorithms, code, and experiments for reproducibility and knowledge transfer. Required Skills - Strong programming in Python; fluency in C++ or solid understanding of C++ for production integration. - Expertise in transformer architectures and deep learning frameworks (e.g., PyTorch, TensorFlow). - Proven experience with large‑scale ML pipelines, including dataset management, training orchestration, and metrics collection. - Solid background in algorithm design and software engineering best practices. - Strong mathematical foundation (probability, statistics, optimization). - Ability to translate research findings into production‑grade solutions. Required Education & Certifications - BS, MS, or PhD in Computer Science, Electrical Engineering, or a related field. - (Optional) Relevant certifications in machine learning or data science are advantageous.

Foster city, United states

Hybrid

Senior

08-09-2025

Company Name: Zoox
Job Title: Machine Learning Engineer - Autotuning
Job Description: Job title: Machine Learning Engineer – Autotuning Role Summary: Design and develop end‑to‑end ML systems that learn from expert human driving data to model and evaluate human‑like autonomous driving behavior, and build AutoML pipelines that optimize driving software. Expectations: - Deliver production‑ready ML solutions that improve vehicle behavior. - Collaborate with cross‑functional teams (planner, simulation, infrastructure). - Work with large‑scale datasets and distributed training environments. Key Responsibilities: - Build ML models that learn human driving patterns from logged data. - Create evaluation metrics to measure the human‑likeness of autonomous driving. - Design and implement AutoML frameworks to tune driving software. - Deploy and maintain ML models at scale, ensuring reliability and performance. - Iterate on data pipelines, feature engineering, and model training workflows. Required Skills: - Proficiency in Python; strong experience with PyTorch or equivalent deep learning libraries. - Background in training, evaluating, and deploying large‑scale ML models. - Familiarity with distributed computing platforms (e.g., Kubernetes, Spark). - Ability to manipulate and analyze extensive behavioral datasets. - Strong problem‑solving and collaboration skills. Required Education & Certifications: - BS/MS/PhD in Machine Learning, Computer Science, or related field (or equivalent professional experience).

Foster city, United states

Hybrid

10-09-2025

Company Name: Zoox
Job Title: Senior Machine Learning Engineer - Perception
Job Description: **Senior Perception Engineer** **Role Summary:** Design and implement cutting-edge detection models using fused sensor data for autonomous vehicle applications. Focus on developing generalizable models to enhance navigation capabilities in diverse environments. **Expactations:** - MS/PhD in Computer Science or related field - Experience with PyTorch or related ML frameworks - Experience deploying learned models into production - Excellent collaboration/teamwork skills with proven large-scale problem-solving track record **Key Responsibilities:** - Train ML models and conduct studies to integrate state-of-the-art (SOTA) approaches or adapted models - Document experimental outcomes and guide deployment into autonomous vehicle systems - Visualize sensor input/output data to diagnose model performance issues - Collaborate with ML optimization, data labeling, metrics, and planning teams for successful project integration **Required Skills:** - ML model training and evaluation techniques - Sensor data integration and analysis - Model deployment and production optimization - Cross-functional teamwork in software/robotics domains **Required Education & Certifications:** - Master’s or PhD in Computer Science, Electrical Engineering, or a related technical discipline

Foster city, United states

Hybrid

Senior

19-10-2025

Company Name: Zoox
Job Title: Machine Learning Engineer, Simulation Scenario Generation
Job Description: **Job Title:** Machine Learning Engineer, Simulation Scenario Generation **Role Summary:** Develop and integrate advanced machine learning models—including large language and vision models—to automate and enhance synthetic scenario creation for autonomous vehicle testing. Drive AI‑based tooling, data collection, and metric design to improve safety validation workflows. **Expactations:** - Deliver production‑ready ML solutions that directly impact internal users. - Collaborate with cross‑functional teams to translate natural‑language test specifications into executable simulation scenarios. - Contribute to model fine‑tuning, performance optimization, and low‑latency inference. **Key Responsibilities:** - Integrate, test, and validate LLM/VLM pipelines and auxiliary models for scenario generation. - Build tooling for AI‑driven scenario understanding, validation, and user feedback. - Design and implement data collection processes and business metrics for model iteration. - Partner with internal customers and engineering teams to embed generative AI into test creation workflows. - Ensure contributions align with safety and reliability standards for autonomous software. **Required Skills:** - Strong proficiency in Python and ML libraries (PyTorch, NumPy). - Deep understanding of LLM/NLP concepts and prompt engineering. - Experience deploying ML models in production environments. - Ability to design experiments, collect data, and evaluate model performance. **Required Education & Certifications:** - MS or PhD in Computer Science, Machine Learning, or a related discipline. - Minimum 2 years of industry experience in machine learning.

Foster city, United states

Hybrid

Junior

21-10-2025

About the Company

Listed Jobs