Senior/Staff Machine Learning Engineer, Agent Simulation

Senior/Staff Machine Learning Engineer, Agent Simulation Location: San Francisco Bay Area Compensation: $200,000 - 300,000 USD / year About The Role The Agent Simulation group at Zoox is searching for machine learning engineers to play a crucial role in enhancing agent behaviors, assessing performance, and developing tools to prioritize simulation results. Your contributions will be key in creating diverse and innovative agent behaviors using real-world driving data and our infrastructure for testing and validating algorithms. In this role, you will: Research, implement, and optimize state-of-the-art machine learning approaches to improve plausible agent behaviors Find innovative solutions for agent behaviors and simulation analysis Prove machine-learned algorithms have better performance than heuristics Leverage our large-scale machine-learning infrastructure to discover new solutions Analyze the difference between behaviors in simulation and real-world Work cross-functionally with our safety and autonomy engineers Build scalable, usable cloud pipelines for machine learning solutions for simulation Qualifications BS, MS, or PhD degree in Computer Science or a related field Experience with training and deploying deep learning models Experience with production machine-learning pipelines: dataset creation, training frameworks, metrics pipelines Fluency in Python and a basic understanding of C++ Fluency with Numpy and PyTorch, TensorFlow or JAX Strong mathematics skills Bonus Qualifications Experience with LLMs, reinforcement learning, imitation learning Conference or journal publications in machine learning or robotics Prior experience with agent behaviors, prediction, autonomous vehicles, or robotics Compensation Salary ranges from $200,000 to $300,000. Compensation also includes Amazon Restricted Stock Units (RSUs) and Zoox Stock Appreciation Rights. Signing bonuses may be part of the package. Compensation varies by location, experience, and related job knowledge. Benefits include paid time off (sick leave, vacation, bereavement), unpaid time off, health insurance, long-term care insurance, long-term and short-term disability insurance, and life insurance. Accommodations If you need an accommodation to participate in the application or interview process, please reach out to View email address on click.appcast.io or your assigned recruiter. #J-18808-Ljbffr