Staff ML Engineer

About the Company

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.

Responsibilities

• Lead and execute large-scale training runs for video or adjacent foundation models
• Contribute to model architecture and training strategy using first-principles understanding
• Improve world-model capabilities for synthetic scenario generation and downstream evaluation
• Partner with research, applications, simulation engineering, and cloud/infrastructure teams
• Provide technical leadership through mentorship, review, and setting high engineering standards

Requirements

• In-depth experience training large-scale models (language, video, or other foundation models)
• Strong understanding of model architecture and training decisions
• Strong hands-on engineering skills with modern ML stacks like PyTorch
• 4–5+ years of relevant industry experience

Preferred Qualifications

• Direct experience with world models, video generation, or long-horizon prediction
• Experience improving data/training pipelines and working across infrastructure constraints
• Proven technical leadership in mentoring and setting direction across an area