Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Staff Machine Learning Engineer, Vision Models

$370.04k - $407.33k
Full-time

Wayve

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. If you’re passionate about self-driving cars and think you have what it takes to make a positive impact on the world, we encourage you to apply.

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.
Vacancy posted 2 days ago
Similar jobs that could be interesting for youBased on the Staff Machine Learning Engineer, Vision Models in California vacancy
  • $203k - $270k

     ...Job Description Job Description Staff Computer Vision & Machine Learning Engineer for Autonomous Anti-Drone Systems Company Overview: Allen Control...  .... Design and implement machine learning models that can operate in resource-constrained environments... 
    Suggested
    Local area

    Allen Control Systems

    Mountain View, CA
    16 days ago
  • $250k - $295k

     ...use advanced physics and AI to model catastrophic risk at the...  ...real product is a scalable risk engine, our Stand World Model . We...  ...The Opportunity: As a Machine Learning Engineer on the Applied Science...  ...heterogeneous data (e.g., 3D/vision, simulation outputs, tabular,... 
    Suggested
    Full time
    Temporary work
    H1b
    Work at office
    Remote work
    Visa sponsorship
    Work visa
    Flexible hours

    Stand Insurance

    San Francisco, CA
    1 day ago
  • $138.91k - $285.98k

     ...development and research of applied machine learning. Our initiatives span a diverse...  ..., including fundamental computer vision, multimodal large language models, multimodal representation...  ...collaborative group of approximately six engineers and a product prototyping team,... 
    Suggested
    Work at office
    Local area
    Remote work
    Relocation
    Relocation package

    Pinterest

    San Francisco, CA
    2 days ago
  • $160k - $250k

     ...and introducing a Factory-as-a-Service model that delivers complete systems instead...  ...for the factory.As a Senior Machine Learning Engineer, you will own the ML lifecycle for the...  ...experience building and shipping computer vision models, with an emphasis on detection... 
    Suggested
    Permanent employment
    Full time
    Local area
    Flexible hours

    Hadrian

    Los Angeles, CA
    3 days ago
  •  ...—including core computer vision, multimodal representation learning, heterogeneous graph neural networks, generative modeling, and recommender systems....  ...pod is a small group (~10 engineers) inside Labs, which...  ...team. M.S. or PhD in Machine Learning or related academic... 
    Suggested
    Full time
    Work experience placement
    Currently hiring
    Work at office
    Remote work
    Relocation
    Relocation package
    Flexible hours

    Pinterest

    San Francisco, CA
    1 day ago
  • $170k - $216k

     ...builds the system which learns the spatial-temporal representation...  ...set of sensors, enabling engineers like you to (1) develop...  ...data, to (2) develop models and model training at...  ...~3+ years experience in Machine Learning and/or Computer Vision ~ Experience with Python... 
    Full time
    Remote work

    Waymo

    San Francisco, CA
    1 day ago
  •  .... Develop and integrate large language models (LLMs) and other state-of-the-art AI techniques...  ...experiences. Knowledge and passion in machine learning algorithms, Gen AI, LLMs, and natural...  ...benefits with family medical, vision and dental coverage, a competitive base... 
    Full time
    Work experience placement

    Eightfold

    Santa Clara, CA
    1 day ago
  • $229k - $343k

     ...themselves, live in the moment, learn about the world, and have...  ...AI, including foundational models, efficient infrastructure, and...  ....We’re looking for a Machine Learning Engineer to join Snap Inc!What you’ll...  ...experienceExperience with Computer Vision or Generative Modeling... 
    Full time
    Live in
    Work at office
    Local area
    Worldwide

    Snap

    Palo Alto, CA
    1 day ago
  • $244.14k - $413.16k

     ...through cutting-edge R&D in AI, machine learning, and smart connectivity.The...  ...we are developing large-scale Vision-Language-Action (VLA) models and World Models to handle the...  ...of global driving. As a Senior Staff Machine Learning Engineer, you will architect the transition... 
    Full time
    Overseas

    XPENG Motors

    Santa Clara, CA
    4 days ago
  • $232k - $310k

     ...collaboration, and high standards. Our engineers, product leaders, and go-to-...  ...the boundaries of applied machine learning. We work with massive...  ...cutting-edge deep learning models across all Eightfold...  ...benefits with family medical, vision and dental coverage, a competitive... 
    Work experience placement
    Work at office
    Remote work
    Flexible hours
    3 days per week

    Eightfold

    Santa Clara, CA
    5 days ago
  • $174.72k - $295.68k

     ...transportation through cutting-edge R&D in AI, machine learning, and smart connectivity.We are...  ...for a full-time Machine Learning Engineer / Research Scientist to drive the modeling and algorithmic development of XPENG’s next-generation Vision-Language-Action (VLA) Foundation... 
    Full time

    XPENG Motors

    Santa Clara, CA
    4 days ago
  • $174.72k - $295.68k

     ...through cutting-edge R&D in AI, machine learning, and smart connectivity. We are seeking Machine Learning Engineers with strong expertise in generative modeling and large-scale deep learning systems...  ...team of experts in computer vision, generative AI, and AI systems, powered... 
    Full time

    XPENG Motors

    Santa Clara, CA
    4 days ago
  • $298k - $368k

     ...builds the system which learns the spatial-temporal...  ...set of sensors, enabling engineers like you to (1) develop...  ...world data, to (2) develop models and model training at...  ...years of experience in Machine Learning, with a focus...  ...learning or computer vision conferences (e.g., NeurIPS... 
    Full time
    Remote work

    Waymo

    San Francisco, CA
    1 day ago
  • $218.4k - $273k

     ...Scale’s robotics offerings, and expand the frontier of Robotics data and model evaluation. We are seeking an exceptionally motivated and experienced Senior Machine Learning Engineer, Computer Vision to drive cutting-edge research and development in real-time and... 
    Full time

    Scale Ai

    San Francisco, CA
    1 day ago
  • $160k - $250k

     ...information for the factory. As a Senior Machine Learning Engineer, you will own the ML lifecycle for the detection and segmentation models at the core of our manufacturing...  ...experience building and shipping computer vision models, with an emphasis on detection and... 
    Permanent employment
    Full time
    Local area
    Relocation package
    Flexible hours

    Hadrian Automation

    Torrance, CA
    10 days ago
  • $138.91k - $285.98k

     ...year Requirements: We look for research engineers or scientists with experience in generative computer vision models, ideally including visual encoders and large...  ...computer vision. We expect an M.S. or PhD in Machine Learning, Computer Science, or a closely related... 
    Work at office
    Remote work
    Worldwide
    Relocation package

    Pinterest, Inc.

    San Francisco, CA
    3 days ago
  • $50.04 - $58.84 per hour

     ...member of the Charging Data Modeling team, you will develop models...  ...algorithms that bridge the engineering, service, deployment, and operation...  ...through rigorous analysis, machine learning modeling, and clear...  ...including orthodontic coverage) and vision plans. Both have an option... 
    Full time
    Temporary work
    Part time
    Internship
    Worldwide
    Flexible hours

    Tesla

    California
    20 days ago
  •  ...autonomous Corvus One ™ drones use computer vision & robotics to automatically track inventory, improving...  ...the role We are hiring Computer Vision / Machine Learning Software Engineers to build compute-constrained models for deployed robots. You'll tackle diverse... 
    Full time
    Visa sponsorship

    Corvus Robotics

    Mountain View, CA
    1 day ago
  • $124k

     ...we’re not just training models, we’re building the...  ...at making massive deep learning models run lightning-fast...  ...foundation models that fuse vision, prediction, and...  ...AI compiler, inference engine, and silicon teams to ensure...  ...in Computer Science, Machine Learning, Robotics, Computer... 
    Hourly pay
    Full time
    Temporary work
    Immediate start
    Flexible hours

    Tesla

    California
    28 days ago
  • $189.31k - $389.75k

     ...Requirements: We are looking for research engineers or scientists with experience building and training computer vision models. We strongly prefer experience with...  ...contributions, and/or experience shipping machine learning models to production. We require practical... 
    Full time
    Work at office
    Worldwide
    Flexible hours

    Pinterest, Inc.

    San Francisco, CA
    3 days ago
  •  ...Institute of Foundation Models  We are a dedicated...  ...data scientists, and engineers, tackling the most fundamental...  ...computing in deep learning, driving impactful...  ...foundation models to unlock machine intelligence beyond...  ...who share our vision and are eager to push... 
    Visa sponsorship

    Institute of Foundation Models

    Sunnyvale, CA
    more than 2 months ago
  •  ...philosophy and how we use AI in our recruiting process here . We are looking for a Staff Machine Learning Engineer to lead the technical vision for our Ads Conversion Core Modeling team, building the state-of-the-art systems that power our global marketplace.... 
    Full time
    Work at office
    Relocation
    Relocation package

    Pinterest

    San Francisco, CA
    1 day ago
  • $94k - $270k

     ...former Navy electrical engineers with a proven track...  ...using advanced computer vision and control systems to...  ...Computer Vision and Machine Learning Engineer to develop and...  ...Associate through Senior Staff; your title and level...  ...machine learning models capable of operating in... 
    Local area

    Allen Control Systems

    Mountain View, CA
    15 days ago
  •  ...easier – this isn’t it. The role We’re looking for a Machine Learning Engineer to build and ship consumer-facing AI systems that power...  ...generation “sleep intelligence.” You’ll work across data, modeling, product, and engineering to translate research into reliable... 
    Full time
    Immediate start
    Worldwide
    Night shift

    Eight Sleep

    San Francisco, CA
    1 day ago
  • $168.1k - $312.3k

     ...leveraging data and novel computational models to drive impact. Seamless data sharing and...  ...Our mission is to develop AI models that learn from biological data at unprecedented scale...  ...motivated and passionate Senior ML Engineer to join our Generative Modeling team and... 
    Full time
    Local area
    Worldwide
    Relocation package

    Genentech

    South San Francisco, CA
    1 day ago
  • $91.6k - $195.6k

     ...house Cell Advanced Technology Engineering group is looking for a highly motivated Machine Learning Engineer. This is a critical role...  ...teams to build intelligent models that drive deeper understanding...  ...including orthodontic coverage) and vision plans, both have options with a... 
    Hourly pay
    Full time
    Temporary work
    Flexible hours

    Jobzhr

    Fremont, CA
    1 day ago
  • $190k - $205k

     ...characteristics. Design models that are robust,...  ...approaches such as graph-based learning for grid topology...  ...signals Production Engineering Write clean,...  ...years of experience in machine learning, signal processing...  ...Benefits Health, Dental & Vision (Gold and Platinum... 
    Full time
    Live in

    Gridware

    San Francisco, CA
    1 day ago
  • $193.3k - $261.5k

     ...AI revolution? At AWS our vision is to make deep learning pervasive for everyday developers...  ...performance of complex ML models executed on AWS Inferentia...  ...is for a senior software engineer in the Compiler team for...  ...employees, supervisors, and staff; adhere to standards of excellence... 
    Full time
    Local area
    Flexible hours

    Annapurna Labs (U.S.) Inc.

    Cupertino, CA
    11 hours ago
  • $160k - $225k

     ...be used to expand our product and engineering teams, bringing our vision of intelligent, autonomous marketing...  ...the manual. As an early Machine Learning Engineer at MAI, you won't just be...  ...-class, 24/7 experience. Bring Models to Life: You will collaborate closely... 
    Full time

    Mai

    Mountain View, CA
    1 day ago
  • $160k - $250k

     ...with a portfolio of best-in-class, pre-trained AI models, serving billions of customer API requests every month...  ...in joining the future of AI! Senior Machine Learning Engineer In order to execute our vision, we need to grow our team of best-in-class machine... 
    Full time

    Hive

    San Francisco, CA
    1 day ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Staff Machine Learning Engineer, Vision Models. Be the first to apply!