Senior Machine Learning Engineer, AV Core
Salary not listed • London, United Kingdom • Full-time
Posted 28d ago
Remote Work Policy
On-site
Employment Type
Full-time
Categories
Machine Learning Engineer
About the job
Wayve is building the leading AI platform for autonomous driving, pioneering an end-to-end AI approach that enables vehicles to learn directly from real-world experience. This role is within the AV Core Engineering Team, specifically the Core Model Safety team, which focuses on collision avoidance, model understanding, and robustness under failure. You will join a senior team working with large-scale training and fleet data, collaborating with research, simulation, evaluation, and applied engineering teams. The core of the work involves leading large-scale data enrichment and curation initiatives, developing workflows for mining hard and rare events, investigating model performance in real-world scenarios, and improving the efficiency and reliability of the machine learning operations stack.
Responsibilities
- Lead large-scale data enrichment within the Core Model Safety team, collaborating with pre/mid-training, reward modelling, self-supervised labelling, and the Data Enrichment Platform.
- Build workflows for large-scale enrichment, data curation, and hard/rare event mining to support robust model performance and introspection.
- Design and train out-of-distribution detection mechanisms for robust driverless operation.
- Work at the intersection of model engineering, infrastructure, compute, and data, shaping team boundaries and architecture.
- Improve the overall efficiency of the machine learning and operations stack using expertise and judgment.
Requirements
- Hands-on experience with ML systems deployed in the real world.
- Proficiency in Python and PyTorch, with strong software engineering practices and experience building reliable ML training and evaluation systems.
- Strong track record with ML operations and infrastructure, including experience with distributed computing environments and large-scale inference.
- Senior-level ownership and collaboration skills, with the ability to lead a substantial technical area and communicate effectively across research and engineering boundaries.
- Prior experience in autonomous vehicles or robotics with hands-on deployment and closed-loop validation on physical systems (desirable).
- Experience mining, generating, or evaluating rare events using simulation and fleet or heterogeneous real-world data (desirable).
- Experience setting up cloud-based monitoring solutions for large-scale operations, including dashboards, logging, and real-time alerts (desirable).
- Experience with transformer-based and multimodal architectures, including vision-language models (VLM), vision-language-action models (VLA), or equivalent (desirable).
- Proficiency in C++, CUDA, distributed training, or performance optimization for production machine learning systems (desirable).
Benefits
- Salaries benchmarked against the market annually
- Meaningful equity, sharing in the ownership and long-term success of Wayve
- Relocation support
- Visa sponsorship where applicable