Staff Machine Learning Engineer, Robotics
Wayve
Sunnyvale, CA · $370k–$420k a year · Posted October 21, 2026
Just laid off? Start here: a plan for the first weeks
More in California job market · Staff Machine Learning Engineer jobs in California (55) · All Staff Machine Learning Engineer jobs (110) · Technology & Software · Data & Analytics jobs
More roles at Wayve
- Lead Technical Program Manager, Data · Sunnyvale, CA
- Principal Research Scientist – Multimodal Reasoning & Embodied AI · Sunnyvale, CA
- Director, AV Product Engineering · Sunnyvale, CA
- Controls Engineer · Sunnyvale, CA
- Staff Technical Product Manager, Robotaxi · Sunnyvale, CA
- Software Engineer, Simulation · Sunnyvale, CA
Similar jobs
- Platform Engineer · Flux · San Francisco, CA
- CANES Platform Engineer · Scientific Research · San Diego, CA
- Systems Engineer II · Crane · Burbank, CA
- Engineer, Rewarming Systems · Until · CA
- Platform Engineer · Sable Bio · San Francisco, CA
- (Sr) Staff SOC Architect - AI Accelerators (Multiple Locations) · Qualcomm Technologies · San Diego, CA
Where Sunnyvale, CA ranks
Where Sunnyvale, CA placed in our monthly rankings, month by month. Each month stays online at its own address.
November 2026
- #1Best cities for consumer and retail pay after cost of living$194k · of 11
- #3Best cities for data and analytics pay after cost of living$213k · of 28
- #4Best cities for pay after cost of living in California$192k · of 32
- #4Best all-round cities for AI careers95 / 100 · of 100
- #4Best all-round cities for AI careers in the West Coast88 / 100 · of 39
- #5Best cities for pay after cost of living in the West Coast$192k · of 39
Rankings about data & analytics jobs
- Best cities for data and analytics jobs #1 New York (1,713 openings)
- Best cities for data and analytics pay after cost of living #1 Foster ($232k)
- Most requested AI skills in data and analytics jobs #1 Machine learning engineering (17.1%)
- AI tools named most often in data and analytics jobs #1 Databricks (7.6%)
- Employers with the most data and analytics jobs #1 Amazon (711 roles)
Data & Analytics jobs · Top Data & Analytics employers · All rankings
How AI is used in this kind of work
Among computer occupations, all other, AI is already used for a large share of the tasks. Measured exposure for computer occupations, all other: 31% (high).
Share of this job’s tasks that AI is already used for at work, from real usage. It is not a forecast of job loss, and a low number means no use seen yet. Source: Anthropic Economic Index.
Skills this job asks for
Say which you have and which you want to build. We use it to match you to jobs like this one.
About the job
Before the detail, here's the challenge you'd help us solve.
We build the embodied intelligence that moves real vehicles safely, and the ecosystem a billion machines will run on in the future. Very few people in AI can say this. Every role here, whatever the team, plugs into that.
Here’s what this particular role covers.
🛠️ ABOUT OUR ENGINEERING TEAMS
Wayve’s engineering teams are building the AI, robotics, simulation, data, and systems foundations needed to deploy a generalisable AI Driver safely and at scale.
ABOUT THE VEHICLE GENERALIZATION TEAM
Vehicle Generalization makes it efficient, scalable, and predictable to deploy Wayve’s autonomous driving stack across a wide range of vehicle platforms by making our core autonomy generalise as widely as possible.
We work across pre- and post-training to improve how well our AI models generalise zero-shot to changes in perception—including cameras, intrinsics, field of view, lens distortion, and calibration—and embodiment, including vehicle dynamics, actuation limits, and rig geometry. We also own the Generalization Benchmark, Wayve’s tool for measuring how models transfer across platforms. Our work goes into production programmes with major OEMs.
We’re looking for a hands-on Machine Learning Engineer with a strong robotics background who understands how a real vehicle senses, estimates its state, and moves. You’ll identify where our models break down across platforms, understand why, and address those failures in training—then prove the improvement through rigorous evaluation.
🧠 YOUR DAY-TO-DAY
- Design and train models that are robust to differences between vehicles, using approaches such as conditioning or augmentation for vehicle kinematics and camera parameters.
- Create benchmarks and metrics for cross-platform performance, translating results into clear decisions for model and platform teams.
- Help root-cause on-road issues such as late turns, lane-keeping, and steering problems. Work with controls, calibration, and OEM engineers to separate model problems from system problems and identify the highest-impact model improvements.
- Move research into production in collaboration with Driving Core and application engineering teams.
- Support customer programmes, with some travel.
- Mentor teammates and help raise the bar for ML engineering across the team.
🧩 WHAT YOU’LL BE WORKING ON
- Pre- and post-training methods that improve zero-shot transfer across vehicle platforms.
- Robustness to changes in camera configuration, calibration, vehicle dynamics, actuation limits, and rig geometry.
- The Generalization Benchmark and metrics that measure how models transfer between platforms.
- Real-world, multi-sensor driving data and the interaction between sensors, data, models, and actuation.
- Production deployment of Wayve’s autonomy stack with major OEM partners.
🙌 YOU SHOULD APPLY IF
- You have several years of experience building and shipping ML systems in PyTorch and Python, including training at scale, debugging experiments, and rigorous evaluation.
- You have a strong robotics or automotive background in areas such as state estimation, sensor fusion, calibration, vehicle dynamics, or controls.
- You bring a systems perspective and can trace a problem through sensors, data, models, and actuation.
- You have experience working with real-world, multi-sensor driving or robot data.
- You communicate clearly across teams and are comfortable owning ambiguous problems from start to finish.
NICE TO HAVE
- Experience with end-to-end or learned driving and planning policies, imitation learning, or reinforcement learning.
- Experience with platform bring-up or deployment across multiple vehicles.
- C++ skills and experience with on-vehicle or embedded deployment.
🌱 Not ticking every box? That’s totally okay! If you’re passionate about autonomy and keen to learn, we encourage you to apply even if you don’t meet every requirement.
MORE ABOUT WAYVE
🚀 Wayve is building the leading AI platform for autonomous driving. We are pioneering an end-to-end AI approach that enables vehicles to learn directly from real-world experience, developing the ability to adapt, generalise and improve at scale.
Our ambition is to make autonomy universal. Wayve’s mapless and hardware-agnostic AI platform integrates with global OEM partners, enabling continuous software evolution from L2+ through to L4.
HOW WE WORK 💻 — LOCATIONS & FLEXIBLE WORKING
This role is based in Sunnyvale, CA (preferred), or London. We operate a hybrid model that combines in-person collaboration with focused time working remotely. Some travel is required to support customer programmes.
🔍 THE INTERVIEW PROCESS
- Initial call / recruiter screen — 25 minutes
- Hiring-manager discussion — 45 minutes
- Deep-dive technical interviews — approximately 4 hours total
- Final interview focused on mission and values alignment — 1 hour
We’ll always explain the format and work around your availability.
WHAT’S IN IT FOR YOU (LOCATION DEPENDENT)
- 💰 Salaries benchmarked against the market annually
- 📈 Meaningful equity
- ✈️ Relocation support and visa sponsorship where applicable
- ✅ Hybrid working and access to vehicle workshops and labs
- 📚 Learning and development support
- 🩺 Comprehensive location-dependent health, family, retirement, and wellbeing benefits
A quick, honest note before you apply.
Wayve is not a mature, fully-structured place with the playbook already written. Much of how we work is still being written, and if you join, you’ll help write it. That suits people who want real ownership more than people who need a settled structure from day one.
If that sounds like the kind of problem you want to spend your time on, we’d really like to hear from you.
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. https://wayve.ai/careers/ To learn more about what drives us, visit Values at Wayve https://wayve.ai/careers/
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.