Machine Learning Jobs

• Lausanne, Switzerland
• Hybrid · 3 days in office
• Senior
• Permanent

Architect and secure an AI-first drug discovery platform, reporting to the CISO. You will build AI/ML threat models, protect model artifacts, deploy LLM and agentic guardrails, and lead AI-specific incident response across distributed training and inference infrastructure.

JAX · PyTorch · TensorFlow +7

• London
• Hybrid · 3 days in office
• Lead · manages people

Lead a team of biologists in the Exploratory Sciences group, owning in vitro pharmacology strategy for novel modalities such as TPD, molecular glues and peptides. Design screening cascades and quantitative assays, set ML-ready data standards, and champion LLM agentic workflows while managing CRO and academic studies.

SPR · GCI · nanoDSF +5

• London
• Hybrid
• Lead · manages people
• Permanent

Lead a small team iterating on Wayve's end-to-end L2 driving model, balancing hands-on Python coding and experiments with directing priorities. Diagnose on-road issues, root-cause them, and drive fixes through data recipes, model architecture, evaluation and release readiness, working across teams on model and vehicle interface problems.

Nissan · Mercedes

• London
• Hybrid
• Lead · manages people
• Permanent

Lead a London-based team building and operating large-scale multi-cloud, multi-region GPU compute and storage infrastructure for ML training and inference. The role is people-first: coaching engineers, hiring, and owning roadmap and delivery, while guiding Kubernetes scheduling, capacity planning and cost trade-offs across heterogeneous GPU fleets.

Kubernetes · Kueue · Nvidia Hopper +2

• Remote (UK)
• Remote
• Lead · manages people
• Permanent

Sets technical direction for complex ML projects in the defence sector, designing and maintaining scalable ML/software systems while justifying architectural decisions. Leads hiring, mentoring and shared library development, working 2-4 days per week on client sites with UK travel and the rest remote or from the London office.

Scikit-learn · TensorFlow · PyTorch +5

• Remote (UK)
• Remote
• Lead · manages people
• Permanent

Lead backend and edge/IoT engineering for high-stakes Defence projects, bridging proof-of-concept work and robust production systems. You'll mentor a squad of engineers, design complex components, and build CI/CD and containerisation pipelines in Python and a compiled language like Rust, C++ or Go, working alongside ML engineers and data scientists. Requires UK Security Clearance eligibility and 2-4 days per week on client sites.

GitLab CI/CD · Docker · Kubernetes +1

• London
• Hybrid
• Permanent

Build and operate backend services, APIs and data-processing workflows for a platform that deploys, runs and evaluates ML models over petabytes of driving data and large GPU fleets. Work spans model-execution workflows, versioned outputs, annotation and dataset-catalogue features, plus reliability, performance and cost work across distributed, compute-intensive systems.

Flyte · Airflow · Dagster +7

• London
• Hybrid
• Mid-level
• Permanent

Builds and operates the CI/CD infrastructure underpinning Wayve's software development lifecycle, maintaining pipelines, build agents, runners and artifact storage. Works with engineering teams to cut build and test feedback times, improving caching, dependency management and reproducible builds. Also develops self-service tooling and automation to scale the build platform.

GitHub Actions · GitLab CI · Jenkins +2

• London
• Hybrid
• Senior
• Permanent

Owns the technical architecture and roadmap for LiDAR and cross-sensor 3D sensing, leading development of target-based and targetless calibration, LiDAR-to-camera alignment, temporal synchronisation and motion compensation. A senior individual-contributor role combining algorithm development, evaluation and production deployment with technical leadership and mentorship across the 3D sensing stack.

GTSAM · Ceres Solver · g2o +1

• London
• Hybrid
• Senior
• Permanent

Lead functional safety activities for an end-to-end autonomous driving system, building Item Definitions, running HARA/FMEA/FTA analyses, and shaping safety concepts and architectures across vehicle platforms and third-party safety-critical systems. Define verification and validation, including vehicle-level testing, and construct structured safety cases with claims, arguments and evidence to support certification.

ISO 26262 · SOTIF

• London
• Hybrid
• Senior
• Permanent

Owns the engineering release process for autonomous driving models, defining branching strategies, release cadence, configuration management and quality gates. Reviews model, architecture and code changes against intended outcomes, assesses evaluation results for safety and performance, and builds CI/CD tooling with PyTorch, TensorRT and GitHub Actions. Works across ML engineering, AI Platform, evaluation and product to unblock delivery bottlenecks.

PyTorch · TensorRT · GitHub Actions +1

• London
• Hybrid
• Lead · manages people
• Permanent

Lead a dedicated Runtime Platform team building the application frameworks, middleware, IPC, performance and observability foundations that support Wayve's onboard AI Driver. The role is people-first, combining coaching, hiring and roadmap ownership with technical direction across latency, profiling, real-time scheduling and embedded Linux, partnering with vehicle software, hardware and OEM programme teams.

QNX · OpenTelemetry · Grafana +7

• London
• Hybrid
• Senior · manages people
• Permanent

Own the technical vision and architecture for a petabyte-scale data enrichment platform that lets model engineers deploy, evaluate and improve ML models across large GPU fleets. Combine hands-on Python backend engineering with cross-team technical leadership, mentoring senior engineers and leading a small team of 2-3.

Flyte · Airflow · Dagster +6

• London
• Hybrid
• Permanent

Develop, integrate and validate control, trajectory-generation and action-processing algorithms that translate AI Driver outputs into safe, smooth vehicle motion across multiple platforms. Own production-quality, latency-sensitive C++ control software, using simulation, closed-loop evaluation and on-road testing to characterise controller behaviour and track comfort, tracking-error and safety KPIs.

CAN

• London
• Hybrid
• Lead
• Permanent

Lead large-scale training and post-training for Gaia, Wayve's world model that predicts future driving video frames and runs closed-loop with the driving model. Own model architecture and training-strategy decisions, push autoregressive generation toward longer, more stable rollouts, and make the model deployment-ready. Work across Research, Applications, Core Simulation Engineering and Cloud/Infrastructure, mentoring and setting technical direction.

PyTorch