Staff Machine Learning Engineer, Vision Models Sunnyvale, California USA

OpenDigital Limited; Sunnyvale, Northern (CA, KY); Hybrid; USD 370,000 - 407,000; Full time

Benefits offered by this job

Equity package

Job summary

Wayve in Sunnyvale is seeking a Staff Machine Learning Engineer to join the Measurement team. You will build the computer vision and scene understanding models Wayve uses to measure offline driving performance and adapt technology from on-vehicle models and foundation models for offline use.

The role requires 5+ years in ML engineering, strong Python and PyTorch skills, and leadership across cross-functional teams. This is a hybrid, full-time role with a competitive equity package.

Qualifications

• 5+ years in ML engineering with production DL model experience.
• Hands-on CV models training with transformer or multimodal architectures.
• Experience fine-tuning large pretrained/foundation models.
• Proficient in Python and ML frameworks (PyTorch).
• Staff-level technical leadership, cross-functional collaboration.
• Ability to define and measure metrics to show improvement.

Responsibilities

• Develop scene understanding models for offline measurement.
• Improve accuracy across platforms, geographies, conditions.
• Utilize offline environment advantages (higher compute, larger models).
• Benchmark models and derive actionable insights.
• Ensure benchmark results are credible for validation pipelines.
• Mentor cross-site teams and align with priorities.

Skills

ML engineering

Computer vision

Python

PyTorch

Technical leadership

Model evaluation

Tools

Distributed training infra

Git

Job description

About us

Founded in 2017, Wayve is the leading developer of Embodied AI technology. Our advanced AI software and foundation models enable vehicles to perceive, understand, and navigate any complex environment, enhancing the usability and safety of automated driving systems.

Our vision is to create autonomy that propels the world forward. Our intelligent, mapless, and hardware-agnostic AI products are designed for automakers, accelerating the transition from assisted to automated driving.

In our fast-paced environment big problems ignite us—we embrace uncertainty, leaning into complex challenges to unlock groundbreaking solutions. We aim high and stay humble in our pursuit of excellence, constantly learning and evolving as we pave the way for a smarter, safer future.

At Wayve, your contributions matter. We value diversity, embrace new perspectives, and foster an inclusive work environment; we back each other to deliver impact.

Make Wayve the experience that defines your career!

The role

As a Staff Machine Learning Engineer on Wayve's Measurement team in AI Evaluation, based in our Sunnyvale office, you will build the computer vision and scene understanding models Wayve uses to measure the performance of the Wayve Driver offline. You will adapt technology from our on-vehicle models and Wayve Foundation Models into offline models that understand coverage, mine rare events, and assess driving behaviour, and you will drive their accuracy and generalisation across vehicles, markets, and conditions. Measuring your own models rigorously is part of the work. You will define ground truth and correctness criteria across a complex driving taxonomy, and turn them into automated benchmarks and evidence that our validation pipelines and safety cases can stand on.

The Measurement team builds and qualifies the scene understanding models Wayve uses to measure driving performance offline, after on-road runs and in simulation. Offline is where the interesting headroom is: more compute per frame, larger foundation models, and access to both past and future temporal context that the vehicle never has. The outputs are mission-critical, directly informing model development decisions and customer deliverables. You will work in a focused, high-impact senior team with strong ownership, access to fleet-scale camera, lidar, and simulation data, and close partners across on-vehicle modelling, evaluation, data curation, and simulation.

Key responsibilities

• 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.
• 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.
• 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.
• 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.
• Align priorities and mentor - work day-to-day with on-vehicle modelling, evaluation, data curation, and simulation teams across sites; raise the bar on engineering and modelling practice; mentor others on the team; keep sight of division and company priorities and how Measurement work enables them.

About you

In order to set you up for success as a Staff Machine Learning Engineer at Wayve, we’re looking for the following skills and experience.

Essential

• 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.
• Hands-on experience training modern computer vision models, including transformer-based and multimodal or VLM architectures for detection, segmentation, classification, or scene understanding, on camera and/or lidar sensor data.
• Experience adapting or fine-tuning large pretrained or foundation models, and training shared representations across multiple tasks or objectives (multi-stage or joint training), including real trade-offs across data and losses.
• Proficient in Python and ML frameworks (esp. PyTorch), with a solid foundation in software engineering practices and comfort with large-scale training.
• 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.

Desirable

• Experience in 3D scene understanding and representation learning for geometric and semantic perception, including large-scale semantic enrichment of driving scenes.
• Experience with offboard or offline modelling: auto-labelling, model distillation, temporal or world models, or other ways of exploiting compute that on-vehicle systems cannot.
• Prior experience in autonomous vehicles or robotics with hands-on deployment and closed-loop validation on physical systems.
• Experience with fleet-scale data and large-scale distributed training infrastructure.

This is a full-time role based in our office in Sunnyvale. At Wayve we want the best of all worlds so we operate a hybrid working policy that combines time together in our offices and workshops to fuel innovation, culture, relationships and learning, and time spent working from home. The reasonably estimated salary for this role ranges from $370,040 to $407,330, plus a competitive equity package. Actual compensation is based on the candidate's skills, qualifications, and experience.

Wayve is committed to creating an inclusive interview experience. If you require any accommodations or adjustments to participate fully in our interview process, please let us know.

We understand that everyone has a unique set of skills and experiences and that not everyone will meet all of the requirements listed above.

At Wayve we're committed to creating a diverse, fair and respectful culture that is inclusive of everyone based on their unique skills and perspectives, and regardless of sex, race, religion or belief, ethnic or national origin, disability, age, citizenship, marital, domestic or civil partnership status, sexual orientation, gender identity, veteran status, pregnancy or related condition (including breastfeeding) or any other basis as protected by applicable law.

For more information visit Careers at Wayve.

To learn more about what drives us, visit Values at Wayve

For US candidates only, please visit E-Verify Notice and Participation and Right to Work

DISCLAIMER: We will not ask about marriage or pregnancy, care responsibilities or disabilities in any of our job adverts or interviews. However, we do look to capture information about care responsibilities, and disabilities among other diversity information as part of an optional DEI Monitoring form to help us identify areas of improvement in our hiring process and ensure that the process is inclusive and non-discriminatory.