Lead ML Engineer, Autonomous Driving & Production

Wayve; United States; Hybrid; USD 89,000 - 121,000; Full time

Benefits offered by this job

Hybrid work policy

Office in Japan

Job summary

Wayve is seeking an ML Engineer to lead model-based autonomous driving initiatives within the Application Engineering team, focusing on core driving performance and feature-level intelligence such as personalization, comfort, and collaboration.

You will design and deliver ML-driven behaviours that scale from assisted to autonomous driving, collaborating with AI Platform, Simulation, Robot SW and Model Release teams to build production-ready systems.

Qualifications

• Extensive and proven track record of shipping deep learning systems to production.
• Expert in deep learning (esp. sequential models, control, planning, or perception).
• Proficient in Python and other relevant languages (e.g. C++ and CUDA) and ML frameworks (esp. PyTorch), with a solid foundation in software engineering practices.
• Experience with real-time systems or robotics, ideally with simulation- or vehicle-in-the-loop components.
• Ability to lead technical initiatives across teams, drive alignment, and mentor engineers.

Responsibilities

• 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, comfort tuning, and user alignment.
• Build evaluation pipelines and metrics for both closed-loop and open-loop driving performance and product readiness.
• Curate and mine real-world and synthetic data to drive scenario diversity, coverage, and feature-specific development.
• Influence architecture choices, training methodologies, and deployment pathways for production-scale learning systems.
• Collaborate cross-functionally across various teams to ensure integration and iteration velocity.
• Mentor senior engineers and shape the long-term technical direction across Autonomy.

Skills

Python

C++/CUDA

PyTorch

Robotics/real-time systems

Technical leadership

Tools

PyTorch

Job description

Wayve is seeking an ML Engineer to lead model-based autonomous driving initiatives within the Application Engineering team, focusing on core driving performance and feature-level intelligence such as personalization, comfort, and collaboration.

You will design and deliver ML-driven behaviours that scale from assisted to autonomous driving, collaborating with AI Platform, Simulation, Robot SW and Model Release teams to build production-ready systems.