GrepJob
Zoox

Senior Machine Learning Engineer - Scene Understanding

Zoox
Apply
about 5 hours ago
Boston, MA, USA or Foster City, CA, USA
Senior / Staff+
H1B Sponsor

Base Salary

$189k - $290k/yr

Responsibilities

  • Design and train Vision-Language-Action (VLA) solutions for robotaxis.
  • Lead end-to-end data strategy, including mining, auto-labeling, and dataset construction.
  • Manage the full post-training stack for VLMs and VLAs, including Continual Pre-training and Supervised Fine-Tuning.
  • Utilize large-scale data pipelines and ML infrastructure to research, prototype, and deploy solutions.
  • Collaborate with cross-functional teams to integrate perception signals.

Requirements

  • MS or PhD in Computer Science or related field.
  • Background in deep learning solutions for VLM and VLA models.
  • Track record in post-training large-scale models, including CPT, SFT, and RL.
  • Hands-on experience with production ML pipelines, including dataset creation and training frameworks.
  • Expertise in Python libraries such as PyTorch, NumPy, Pandas, and VLLM.

Tech Stack

NumPyPandasPythonPyTorch

Categories

AI & MLData Science