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
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
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
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
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
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.