Staff / Senior Machine Learning Engineer, Reinforcement Learning

Job summary

Responsibilities

As a Senior / Staff Machine Learning Engineer in Wayve's AV Core organisation, you will advance reinforcement learning methods for end-to-end driving models. You'll identify where learning from reward or feedback can improve beyond behavior cloning, then take promising ideas from design through…

• Shape and execute the reinforcement learning roadmap for Driving Core / Core Model Safety, selecting problems and methods against clear behavioral gaps and measurable success…
• Develop and evaluate post-behavior-cloning optimization methods, including offline and off-policy reinforcement learning as well as other reward-guided approaches; design the…
• Help improve the reward models and related learning signals used to train and evaluate driving policies, working with partner teams to strengthen their quality, scalability, and…
• Build robust training and experimentation workflows using large-scale driving data; diagnose distribution shift, objective misspecification, optimization instability, and data or…
• Define evidence across offline metrics, open-loop tests, closed-loop simulation, and on-road evaluation, and distinguish genuine policy improvement from benchmark overfitting.

Tech stack

• Python
• PyTorch
• C++
• CUDA