Skip to content

Senior / Staff Machine Learning Engineer - Perception Attributes

242k – 333kFoster City, CAHybrid7+ YOE
Summary

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.

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 Models (VLMs)Vision Transformers (ViTs)C++Data LabelingAuto-labelingDataset Creation
Similar roles at this salary range
All ML Engineering jobs →
Airbnb

Senior Staff Machine Learning Engineer, Communication & Connectivity

Lead ML architecture and implementation for Airbnb's Messaging & Notifications, building recommendation engines, ranking systems, and LLM-powered experiences while mentoring engineers.

244k – 305kUnited StatesML EngineeringRemotePythonAI Systems
Traba

Staff Software Engineer

Founding Staff Applied Agent Engineer to architect and lead Traba's agentic platform, building production LLM/agent systems that integrate with customer WMS/TMS/ERP and drive industrial operations. Requires 7+ years engineering experience with 2+ years building production agent systems.

240k – 300kNew York, NY +1ML EngineeringOn-siteLLMKafka
Traba

Senior Software Engineer

Founding Senior Applied Agent Engineer building production LLM agent systems that automate supply chain workflows. Requires 5+ years engineering experience with 1+ year shipping LLM/agent features, strong Python/TypeScript skills, and hands-on agent stack experience.

200k – 240kNew York, NY +1ML EngineeringOn-sitePythonNode.js
Cribl

Staff Software Engineer, Cribl AI

Staff-level AI/ML engineer building and productionizing generative AI features across backend and frontend for Cribl's observability platform. Requires 6+ years experience, AI/ML and MLOps background, and TypeScript/JavaScript proficiency.

225k – 265kUnited StatesML EngineeringRemoteLLMsReact
Perplexity

Member of Technical Staff

ML Engineer building and optimizing production recommendation, ranking, and personalization systems that integrate LLMs for Perplexity's AI product.

220k – 405kSan Francisco, CA +1ML EngineeringOn-siteLLMsFeature Stores