Staff / Senior Machine Learning Engineer, Reinforcement Learning

📋 Description

• Shape and execute the reinforcement learning roadmap for Driving Core / Core Model Safety
• Develop and evaluate post-behavior-cloning optimization methods, including offline and off-policy
• Help improve the reward models and related learning signals used to train and evaluate driving
• Build robust training and experimentation workflows using large-scale driving data; diagnose
• Define evidence across offline metrics, open-loop tests, closed-loop simulation, and on-road
• Productionize successful methods in the shared ML stack, communicate decisions and results clearly

🎯 Requirements

• You have a strong track record developing and experimentally validating reinforcement learning or
• You have a deep understanding of modern reinforcement learning fundamentals, including policy and
• You have hands-on experience with behaviour cloning, reinforcement learning, or related methods.
• You're proficient in Python and PyTorch, with strong software engineering practices and hands-on
• You have excellent experimental judgement: able to turn an ambiguous behavioral problem into
• You bring senior-level ownership and collaboration: able to lead a substantial technical area, work

🎁 Benefits

• Salaries benchmarked against the market annually.
• Meaningful equity, sharing in the ownership and long term success of Wayve.
• Relocation support and visa sponsorship where applicable.
• Hybrid working, core hours and the chance to work hands on in vehicle workshops and labs.
• Learning and development budgets with support for training, conferences and growth.
• Comprehensive benefits including health insurance, dental, enhanced maternity and paternity leave