Software Engineer (ML Platform)



Software Engineering, Data Science
London, UK
Posted on Thursday, May 16, 2024

Who we are

Established in 2017, Wayve is a leader in autonomous vehicle technology, driven by breakthroughs in Embodied AI. Our intelligent, mapless, and hardware-agnostic technologies empower vehicles to navigate complex environments effortlessly.

Supported by prominent investors, Wayve is advancing the transition from assisted to fully automated driving, making transportation safer, more efficient, and universally accessible. Join our world-class, multinational team of engineers and researchers as we push the boundaries of frontier AI and autonomous driving, creating impactful technologies and products on a global scale

Where you will have an impact

We’re looking for bold, talented and creative people to join our journey in developing next-generation autonomous vehicles. We’re a growing start-up, building our first cohort of engineers and you can be at the heart of this.

You will be responsible for designing, deploying, and maintaining the infrastructure and software that support the team's models training and related applications for both the engineering and the research teams. You will be in charge of optimising training workloads to make machine learning training more efficient and more cost effective using both platform and software optimizations.

We are looking for Software Engineers to help build the Wayve Machine Learning platform.

Challenges you will own

  • The ML Platform team owns the machine learning training infrastructure and works with users to ensure that this infrastructure is reliable and efficiently utilised.
  • You will be part of a growing group focussed on making training infrastructure available to users, for distributed training of large models.
  • You will be working across functions with machine learning research engineers to optimise models so that they can be trained efficiently, saving both money and researcher time.
  • You will have opportunities to develop new skills, especially in model optimisation.

Examples Projects:

  • Working with machine learning researchers to optimise ML models, using the latest tooling like NVIDIA NSight.
  • Training job scheduling and orchestration e.g. tooling to schedule long running jobs at off-peak times.
  • Tooling which provides thousands of GPUs simultaneously to our driving simulator, which we use to test the driving performance of our models off road.

We understand that everyone has a unique set of skills and experiences and that not everyone will meet all of the requirements listed below. If you’re passionate about self-driving cars and think you have what it takes to make a positive impact on the world, we encourage you to apply.

What you will bring to Wayve


  • Proficiency in Python
  • Knowledge of software engineering practices - what makes code reusable and extensible.
  • Experience working with concurrent, parallel and distributed computing.
  • Passion for infrastructure: building internal tooling and frameworks.
  • Experience with cloud infrastructure, preferably Azure
  • Experience with Docker, Kubernetes and Terraform


  • Experience profiling and optimising ML models e.g. with NVIDIA NSight.
  • Experience working with at least one ML framework e.g. Pytorch, Tensorflow, ONNX and TensorRT

What we offer you

  • The chance to be part of a truly mission driven organisation and an opportunity to shape the future of autonomous driving. Unlike our competitors, Wayve is still relatively small and nimble, giving you the chance to make a huge impact

  • Competitive compensation and benefits

  • A dynamic and fast-paced work environment in which you will grow every day - learning on the job, from the brightest minds in our space, and with support for more formal learning opportunities too

  • A culture that is ego-free, respectful and welcoming (of you and your dog) - we even eat lunch together every day

  • Benefits such as an onsite chef, workplace nursery scheme, private health insurance, cycle scheme, therapy, yoga, two onsite bars, large social budgets

  • This is a full-time role based in our office in London. At Wayve we want the best of all worlds so we operate a hybrid working policy that combines time together in our offices and workshops to fuel innovation, culture, relationships and learning, and time spent working from home. We also operate core working hours so you can be where you need to be for family and loved ones too. Teams determine the routines that work best for them.

At Wayve we're committed to creating a diverse, fair and respectful culture that is inclusive of everyone based on their unique skills and perspectives, and regardless of gender, gender identity, gender expression, race, sexual orientation, physical or mental disability, ethnicity, age or religious belief.