GrepJob
Zoox

Machine Learning Engineer - Simulation Framework

Zoox
Apply
about 5 hours ago
Foster City, CA, USA or Seattle, WA, USAMid Level / Senior
H1B Sponsor

Base Salary

$151k - $257k/yr

Responsibilities

  • Develop and optimize GPU-based simulation frameworks for ML training.
  • Apply reinforcement learning to solve behavioral and path planning challenges.
  • Identify and resolve fidelity gaps between simulations and real-world execution.
  • Build systems for autonomy users to self-serve data generation.
  • Write robust, production-ready code for ML algorithm integration.

Requirements

  • PhD or Master’s in computer science, robotics, machine learning, or related field.
  • Deep understanding of reinforcement learning in simulated environments.
  • Hands-on experience with deep learning models using frameworks like JAX or PyTorch.
  • Strong proficiency in C++ and Python for production ML systems.
  • Experience analyzing fidelity gaps between synthetic data and real-world execution.

Categories

AI & MLData Science