Skip to content
ZooxZooxFoster City, CA

Senior / Staff Machine Learning Engineer - Perception Attributes

Develops and maintains behavioral models for road users (vehicles, pedestrians, cyclists) in autonomous driving perception stack. Requires MS/PhD, 7+ years experience, deep learning expertise, Python fluency, and production ML pipelines.

242k – 333k
Hybrid7+ YOEML Engineering

About the role

Responsibilities

  • Lead the development of sophisticated behavioral models for vehicles, pedestrians, and cyclists as a key member of the Attributes team within Zoox's Perception department.
  • Create and maintain perception attribute models that generate essential signals, enabling our autonomous vehicles to understand and predict the behavior of various road users.
  • Collaborate closely with Prediction and Planning teams to optimize your models' outputs, directly influencing how our autonomous vehicles make real-time driving decisions.
  • Work with data labeling and ontology teams on data labeling and ontology definitions of the road users in different attributes and generate auto-labeling or data mining strategies for different attributes.

Qualifications

  • MS/PhD in computer science or related fields with a minimum of 7 years of relevant experience
  • Experience with training and deploying Deep Learning models
  • Experience with knowledge distillations from large foundation models
  • Experience with production Machine Learning pipelines: dataset creation, training frameworks, metrics pipelines
  • Fluency programming in Python and extensive experience with algorithm design
  • Strong mathematics skills

Bonus Qualifications

  • Familiarity of VLMs/VLAs/ViTs
  • Experience with large model distillation in a production environment
  • Familiarity with C++

Skills

Deep LearningMachine Learning PipelinesPythonKnowledge DistillationVision Language ModelsVlmsVision TransformersVitsC++Data LabelingAuto-LabelingDataset Creation

Similar roles

ML Engineering jobs
Zoox

Senior/Staff Machine Learning Engineer - Offline Driving Intelligence

ZooxFoster City, CA +2

Leads ML initiatives to enhance validation processes using large-scale fleet and synthetic data for autonomous vehicle safety. Requires PhD or 5+ years ML experience, Python/PyTorch expertise, and AV domain knowledge.

242k – 333k
Hybrid5+ YOEML Engineering
Zoox

Staff Software Engineer, ML Performance Optimization

ZooxFoster City, CA +1

Leads ML performance optimization for training and inference platforms in autonomous driving, collaborating across teams to enhance efficiency using PyTorch, TensorRT, and profiling tools. Requires strong Python/C++ skills, GPU expertise, and 4+ years in large-scale ML platforms; leads engineering team.

242k – 389k
On-site4+ YOEML Engineering
Traba

Staff Software Engineer

TrabaNew York, NY +1

Build and scale Traba's applied AI platform for industrial supply chain staffing and operations. Architect core backend systems integrating frontier models and agents for automation, matching, and vetting while owning performance and infrastructure.

240k – 300k
Hybrid7+ YOEML Engineering
Traba

Staff Software Engineer

TrabaNew York, NY +1

Build and lead Traba's agentic AI platform as a founding member of the Agents team. Architect orchestration, evals, model strategy, and integrations for autonomous agents in industrial supply chain workflows. Requires 7+ years engineering experience including production LLM/agent systems, with customer immersion and 0-to-1 build experience.

240k – 300k
Hybrid7+ YOEML Engineering
Airbnb

Senior Staff Machine Learning Engineer, Trust

AirbnbUnited States

Senior technical leader building, productionizing and operating large-scale ML models and Agentic AI systems to fight fraud, ensure safety and build trust across Airbnb's platform. Requires 12+ years applied ML experience and deep expertise in LLMs/GenAI.

244k – 305k
Remote12+ YOEML Engineering