Machine Learning Scientist/Engineer
$311k - $420k • Sunnyvale, California USA • Full-time
Posted 28d ago
Job Location
Sunnyvale, California USA
Tech Stack
Remote Work Policy
On-site
Employment Type
Full-time
Categories
AI Research Engineer
About the job
Wayve is building the leading AI platform for autonomous driving, pioneering an end-to-end AI approach that enables vehicles to learn directly from real-world experience. This role is for a founding member of the Multi-Embodiment Generalist Agent (MEGA) team, focused on building foundation models for general-purpose robots. These models will learn from large-scale video, language, and robot-interaction data to generalize across tasks and embodiments, enabling agents to perceive, reason about, and act reliably in the physical world. The work combines frontier ML research with direct real-world evaluation on a growing fleet of robots.
Responsibilities
- Design and run experiments on model architectures, learning objectives, and data strategies for robot foundation models.
- Build scalable pre-training and post-training methods using web video, egocentric video, and robot-interaction data.
- Curate, filter, mix, and evaluate large robotics datasets, including egocentric, UMI, and teleoperated data.
- Develop and use distributed training pipelines for large multimodal models and datasets.
- Partner with ML engineers, roboticists, and hardware teams to translate research progress into measurable real-world robot performance.
- Communicate findings through rigorous internal reviews, publications, and robot demonstrations.
Requirements
- Experience in machine learning, with a focus on 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 (e.g., ICRA, CoRL, CVPR, NeurIPS, ICML, 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.