Staff Machine Learning Engineer, Vision Models

Wayve Wayve · Robotics · Sunnyvale, CA · Simulation, Evaluation, Validation

Staff Machine Learning Engineer on the Measurement team, focusing on building and adapting computer vision and scene understanding models for offline performance measurement of Wayve's autonomous driving systems. The role involves developing, training, and fine-tuning models, driving their accuracy and generalization, and rigorously measuring their performance using defined ground truth and correctness criteria. This position leverages fleet-scale data and adapts foundation models for offline analysis, directly influencing model development decisions and safety cases.

What you'd actually do

  1. Develop the models - build, train, and fine-tune the scene understanding models at the centre of Wayve's offline measurement, adapting on-vehicle architectures and Wayve Foundation Models for offline use.
  2. Drive accuracy and generalisation - improve model performance across vehicle platforms, geographies, and driving conditions; diagnose failure modes and close the loop on blind spots.
  3. Exploit the offline environment - use the advantages the vehicle does not have: higher compute budgets, larger model capacity, bidirectional temporal context, and multi-task or joint representation learning.
  4. Measure what you build - benchmark your models, set quality bars, and use metrics and error analysis to steer the next iteration; treat measurement as the feedback that drives the modelling.
  5. Make the evidence credible - ensure benchmarked results are statistically defensible and fit to feed validation pipelines at scale and our broader safety cases, across the product portfolio.

Skills

Required

  • 5+ years in ML engineering
  • training and shipping deep learning models in production
  • pathfinding in ambiguous modelling problems
  • training modern computer vision models
  • transformer-based and multimodal or VLM architectures
  • detection, segmentation, classification, or scene understanding
  • camera and/or lidar sensor data
  • adapting or fine-tuning large pretrained or foundation models
  • training shared representations across multiple tasks or objectives
  • Python
  • ML frameworks (esp. PyTorch)
  • software engineering practices
  • large-scale training
  • Staff-level technical leadership
  • setting direction
  • raising the bar
  • leading cross-functional work
  • measuring your own models
  • defining and reading metrics

Nice to have

  • 3D scene understanding
  • representation learning for geometric and semantic perception
  • large-scale semantic enrichment of driving scenes
  • offboard or offline modelling
  • auto-labelling
  • model distillation
  • temporal or world models
  • autonomous vehicles or robotics
  • hands-on deployment and closed-loop validation on physical systems
  • fleet-scale data
  • large-scale distributed training infrastructure

What the JD emphasized

  • 5+ years in ML engineering, including training and shipping deep learning models in production, with pathfinding in ambiguous modelling problems from scoping through to a direction others build on.
  • Staff-level technical leadership: research-literate and pragmatic, setting direction, raising the bar, and leading cross-functional work without formal line management.
  • Able to measure your own models: comfortable defining and reading the metrics that show whether a model is genuinely improving.

Other signals

  • Develop the models - build, train, and fine-tune the scene understanding models at the centre of Wayve's offline measurement, adapting on-vehicle architectures and Wayve Foundation Models for offline use.
  • Drive accuracy and generalisation - improve model performance across vehicle platforms, geographies, and driving conditions; diagnose failure modes and close the loop on blind spots.
  • Measure what you build - benchmark your models, set quality bars, and use metrics and error analysis to steer the next iteration; treat measurement as the feedback that drives the modelling.