Machine Learning Engineer, App SW

About this role

Wayve, a leading autonomous driving AI company, seeks an ML Engineer to develop advanced driving models and AI-driven behaviors for both assisted and fully autonomous vehicles. You'll design ML systems spanning architecture, data pipelines, and real-world deployment while collaborating across teams to scale from research to production.

What you'll do

• Develop and improve end-to-end driving models with state-of-the-art performance, robustness, and generalization
• Lead projects on personalized and collaborative driving, including behavior conditioning and comfort tuning
• Build evaluation pipelines and metrics for closed-loop and open-loop driving performance
• Curate and mine real-world and synthetic data to drive scenario diversity and feature development
• Influence architecture choices and training methodologies for production-scale learning systems
• Mentor senior engineers and shape long-term technical direction across the Autonomy team

What they're looking for

• Deep learning (sequential models, control, planning, perception)
• Python, C++, CUDA
• PyTorch and ML frameworks
• Production deep learning systems deployment
• Real-time systems and robotics
• Simulation and vehicle-in-the-loop development
• Technical leadership and cross-functional collaboration
• Autonomous driving or imitation learning (desirable)

Benefits

• Competitive salary ($283,500–$381,600 range)
• Equity package
• Hybrid work policy with flexible scheduling
• Core working hours allowing schedule flexibility
• Inclusive, diverse workplace culture
• Opportunity to work on cutting-edge autonomous vehicle technology