Staff Machine Learning Scientist/Engineer

Wayve

Sunnyvale (CA)

Hybrid

USD 370,000 - 419,000

Full time

45 hours ago
Be an early applicant
Application generator

Don’t send a generic resume — generate a resume and cover letter tailored to this exact role.

Get past ATS filters

Benefits offered by this job

Competitive equity package

Job summary

Wayve is seeking a Research Scientist to join the MEGA team in Sunnyvale, CA, building foundation models for general-purpose robots. You will help define the foundation-model learning stack and collaborate across ML, robotics, and hardware teams to translate research into real-world demonstrations.

Key work includes novel architectures, pre-training objectives, and scalable training systems. The role offers hybrid work with a growing fleet of robots for evaluation and real-world impact.

Qualifications

  • Experience with multimodal foundation models and data for foundation models.
  • Experience with scalable training across multiple nodes and large datasets.
  • Strong publication record in top-tier venues such as ICRA, CoRL, CVPR, NeurIPS, ICML or ICLR.
  • Strong engineering skills and hands-on experience with modern ML frameworks.

Responsibilities

  • Research and develop model architectures, learning objectives, and data strategies for robot foundation models.
  • Empirical research experience – experience hill climbing on ML models.
  • Experience with various data sources – annotation, filtering, mixing strategies.
  • Develop scalable self-supervised and generative pre-training methods using web video, egocentric video, and robot-interaction data.
  • Develop post-training approaches—including supervised fine-tuning, imitation learning, reinforcement learning, and related methods—to improve real-world robot capabilities.
  • Curate, filter, and evaluate large-scale robotics datasets, including egocentric, UMI, and teleoperated data.
  • Build and use distributed training pipelines for large models and large multimodal datasets.
  • Work closely with robotics and hardware teams to connect model progress to measurable real-world performance.

Skills

Multimodal foundation models
Scalable training
Publications in top venues
Engineering skills with ML frameworks
Experiment design
Cross-team collaboration
Communication

Education

PhD or MS in CS/ML/Robotics/Vision

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.

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

We are looking for a Research Scientist to join the Multi-Embodiment Generalist Agent (MEGA) team within Wayve Science as a founding member.

MEGA is building foundation models for general-purpose robots: models that learn from large-scale video, language, and robot-interaction data, then generalize across tasks and embodiments—including mobile manipulators, dual-arm platforms, and humanoids. Our aim is to build agents that can perceive, reason about, and act reliably in the physical world.

You will help define and build the foundation-model learning stack for robotics: novel model architectures, pre-training objectives, post-training methods, and scalable data and training systems. The work combines frontier ML research with a direct route to real-world evaluation on a growing fleet of robots.

Your work may span vision-language-action models, world and action models, video and multimodal models, imitation learning, reinforcement learning, and self-supervised learning. You will work with large-scale video and robotics datasets and distributed training infrastructure to develop increasingly capable, robust, and general robot policies.

You will collaborate with research scientists, ML engineers, roboticists, and hardware teams to turn promising ideas into large-scale experiments, strong research contributions, and compelling robot demonstrations. This is an opportunity to take meaningful ownership of a new ML-first research program at the frontier of foundation models and embodied intelligence.

Key Responsibilities
  • Research and develop model architectures, learning objectives, and data strategies for robot foundation models.
  • Empirical research experience – experience hill climbing on ML models.
  • Experience with various data sources – annotation, filtering, mixing strategies.
  • Develop scalable self-supervised and generative pre-training methods using web video, egocentric video, and robot-interaction data.
  • Develop post-training approaches—including supervised fine-tuning, imitation learning, reinforcement learning, and related methods—to improve real-world robot capabilities.
  • Curate, filter, and evaluate large-scale robotics datasets, including egocentric, UMI, and teleoperated data.
  • Build and use distributed training pipelines for large models and large multimodal datasets.
  • Work closely with robotics and hardware teams to connect model progress to measurable real-world performance.
  • Communicate research clearly internally and, where appropriate, through publications and Wayve’s scientific presence.
About You

In order to set you up for success as a Research Scientist at Wayve, we’re looking for the following skills and experience.

Essential
  • Experience in machine learning, with focus in multimodal foundation models and data for foundation models.
  • Experience with scalable training, such as multi-node training, large datasets and/or large model training.
  • Strong research track record, including publications in top-tier venues such as ICRA, CoRL, CVPR, NeurIPS, ICML or ICLR.
  • Strong engineering skills and hands-on experience with modern machine learning frameworks.
  • Ability to design and run rigorous experiments while collaborating closely with engineering and robotics teams.
  • Experience translating research ideas into working systems, experiments or deployed capabilities.
  • Strong communication skills and the ability to share research clearly across teams
Desirable
  • PhD or MS in Computer Science, Machine Learning, Robotics, Computer Vision or a related technical field.
  • Industry experience in machine learning, robotics, embodied AI or related applied research environments.
  • Experience with real robots, robotic learning, embodied AI, simulation or policy learning.
  • Experience working with large-scale video data and sequential decision-making systems.

This role is a full-time role based in Sunnyvale, CA (hybrid) and the reasonably estimated salary for this role ranges from $370,000 to $419,000.

Benefits
  • plus a competitive equity package.

Actual compensation is based on the candidate's skills, qualifications, and experience. 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. We operate core working hours so you can determine the schedule that works best for you and your team.

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.

Get your free, confidential resume review.
or drag and drop your file here.
Similar jobs

Similar jobs worth comparing

Staff Machine Learning Scientist/Engineer
Staff Machine Learning Scientist/Engineer

Socket.dev • Sunnyvale (CA)

Hybrid
USD 370,000 - 419,000
Staff Machine Learning Software Engineer
Staff Machine Learning Software Engineer

Wayve • Sunnyvale (CA)

Hybrid
USD 311,000 - 419,000
Roboticist, Robot Foundation Model
Roboticist, Robot Foundation Model

Wayve • Sunnyvale (CA)

Hybrid
USD 184,000 - 240,000
Equity package
Hybrid work policy
Staff Machine Learning Software Engineer
Staff Machine Learning Software Engineer

Icehouseventures • Sunnyvale (CA)

Hybrid
USD 311,000 - 419,000
Equity package
Research Scientist, Wayve Labs
Research Scientist, Wayve Labs

Icehouseventures • Sunnyvale (CA)

Hybrid
USD 140,000 - 180,000
Salary and equity
Onsite chef
Private health insurance
+2
Roboticist, Robot Foundation Model
Roboticist, Robot Foundation Model

Socket.dev • Sunnyvale (CA)

Hybrid
USD 184,000 - 240,000
Equity package
Applied Scientist / Machine Learning Engineer
Applied Scientist / Machine Learning Engineer

Icehouseventures • Sunnyvale (CA)

Hybrid
USD 311,000 - 370,000
Principal Engineer, Model Development Platform
Principal Engineer, Model Development Platform

Icehouseventures • Sunnyvale (CA)

Hybrid
USD 295,000 - 336,000
Research Scientist, Robot Foundation Model
Research Scientist, Robot Foundation Model

EngineersOfAI • Sunnyvale (CA)

On-site
USD 180,000 - 240,000
Senior Machine Learning Engineer, Vision Models
Senior Machine Learning Engineer, Vision Models

Wayve • Sunnyvale (CA)

Hybrid
USD 312,000 - 370,000
Competitive equity package