Software Engineer

Wayve; Greater London; On-site; GBP 90,000 - 135,000; Full time

Job summary

Wayve in London is seeking a Software Engineer for the AI Libraries team. You will build and maintain platforms, libraries, and tools that enable Wayve’s ML engineers and researchers to train, evaluate, and scale models efficiently.

You’ll design stable, modular systems, collaborate with ML teams, and improve the reliability and performance of training infrastructure at scale. This role focuses on software engineering over ML modelling, delivering practical, scalable solutions.

Qualifications

• Proficient Python programming with ability to design and deliver software systems.
• Strong software architecture skills with experience building reusable tools or libraries.
• Experience with cloud environments, preferably Azure.
• Experience with concurrent, parallel, or distributed computing.
• Familiarity with ML frameworks such as PyTorch or TensorFlow.
• Ability to work with technical stakeholders to refine requirements and deliver scalable solutions.

Responsibilities

• Design, build, and maintain scalable Python libraries and tools for ML engineers and researchers.
• Develop robust abstractions for data loading, distributed training, inference, and evaluation workflows.
• Support training at scale across large GPU clusters and cloud infrastructure.
• Collaborate with ML teams to understand needs and create reliable, well-documented tools.
• Improve software architecture, testing, monitoring, and maintainability of ML systems.

Skills

Python

Software architecture

Distributed computing

ML frameworks

Stakeholder collaboration

Tools

Azure

Docker

Kubernetes

Prometheus

Grafana

Datadog

OpenTelemetry

Job description

About Us

Founded in 2017, Wayve is the leading developer of Embodied AI technology. Our advanced AI software and foundation models enable vehicles to perceive, understand, and navigate any complex environment, enhancing the usability and safety of automated driving systems.

Our vision is to create autonomy that propels the world forward. Our intelligent, mapless, and hardware-agnostic AI products are designed for automakers, accelerating the transition from assisted to automated driving.

In our fast-paced environment big problems ignite us—we embrace uncertainty, leaning into complex challenges to unlock groundbreaking solutions. We aim high and stay humble in our pursuit of excellence, constantly learning and evolving as we pave the way for a smarter, safer future.

At Wayve, your contributions matter. We value diversity, embrace new perspectives, and foster an inclusive work environment; we back each other to deliver impact.

Make Wayve the experience that defines your career!

The role

We’re looking for a Software Engineer to join our AI Libraries team. This team builds and maintains the platforms, libraries, and tools that enable Wayve’s ML engineers and researchers to train, evaluate, and scale models efficiently.

This is a hands-on software engineering role focused on building stable, scalable, and modular systems that support large-scale ML development. You’ll work closely with ML teams across Wayve to understand their needs, design reusable abstractions, and improve the reliability, performance, and usability of our training infrastructure.

You’ll play a key role in maturing Wayve’s AI platform and helping bring autonomous driving technology into the hands of customers.

Key Responsibilities

• Design, build, and maintain scalable Python libraries and tools used by ML engineers and researchers across Wayve.
• Develop robust abstractions for data loading, distributed training, inference, checkpointing, and model evaluation workflows.
• Support training at scale across large GPU clusters and cloud-based infrastructure.
• Work closely with ML teams to understand user needs and create tools that are reliable, well-documented, observable, and easy to adopt.
• Improve engineering quality across ML systems through strong software architecture, testing, monitoring, and maintainability practices.
• Optimise data and training pipelines to support multi-modal data sources, including camera, radar, lidar, and other sensor data.
• Contribute to the evolution of Wayve’s AI platform as we scale our autonomous driving capabilities.

About You

We’re looking for a strong software engineer who enjoys building high-quality tools, platforms, and libraries for technical users. You care about clean abstractions, scalable architecture, reliability, and creating software that other engineers can depend on.

Essential Skills

• Strong Python programming experience. Proven experience designing, building, and maintaining software systems from concept through to delivery.
• Strong software architecture and system design skills. Experience building tools, platforms, or libraries for internal or external users. Strong understanding of testing, observability, maintainability, and engineering best practices.
• Experience working with cloud environments, ideally Azure.
• Experience with concurrent, parallel, or distributed computing.
• Familiarity with ML frameworks such as PyTorch, TensorFlow, or PyTorch Lightning.
• Ability to work closely with technical stakeholders to refine requirements and deliver practical, scalable solutions.

Desirable Skills

• Experience working with large GPU clusters or distributed training environments.
• Familiarity with distributed training techniques such as DDP or FSDP.
• Experience with observability tools such as Prometheus, Grafana, Datadog, or OpenTelemetry.
• Experience with data pipeline orchestration tools such as Airflow, Flyte, Ray, Metaflow, or Argo Workflows.
• Experience with containerisation and infrastructure tooling such as Docker, Kubernetes, or Terraform. Experience profiling or optimising ML systems, for example using NVIDIA Nsight.
• Understanding of ML workflows and researcher experience, even if you are not focused on model development.

What We’re Not Looking For

This is not primarily an ML modelling role. While an understanding of ML workflows is valuable, the core focus is on building reliable software, libraries, infrastructure, and tooling that enable ML teams to work effectively at scale.

Why join us?

• Work on high-impact systems that directly support the development of autonomous driving technology.
• Help scale training and evaluation infrastructure across large GPU clusters.
• Build software used by ML engineers and researchers working at the frontier of embodied AI.
• Join a team focused on strong engineering standards, practical abstractions, and scalable platform design.
• Play a meaningful role in bringing autonomous driving technology closer to real-world deployment.

Wayve is committed to creating an inclusive interview experience. If you require any accommodations or adjustments to participate fully in our interview process, please let us know.

We understand that everyone has a unique set of skills and experiences and that not everyone will meet all of the requirements listed above.

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 sex, race, religion or belief, ethnic or national origin, disability, age, citizenship, marital, domestic or civil partnership status, sexual orientation, gender identity, veteran status, pregnancy or related condition (including breastfeeding) or any other basis as protected by applicable law.

For more information visit Careers at Wayve.

To learn more about what drives us, visit Values at Wayve

For US candidates only, please visit E-Verify Notice and Participation and Right to Work

DISCLAIMER: We will not ask about marriage or pregnancy, care responsibilities or disabilities in any of our job adverts or interviews. However, we do look to capture information about care responsibilities, and disabilities among other diversity information as part of an optional DEI Monitoring form to help us identify areas of improvement in our hiring process and ensure that the process is inclusive and non-discriminatory.