Machine Learning Jobs

• London
• Hybrid · 3 days in office
• Senior

Research Engineers build and optimise state-of-the-art AI models for drug design, working across the full ML lifecycle from experimentation and evaluation to production deployment. The role involves developing robust codebases, data pipelines and training infrastructure, with a focus on architectures such as Transformers, GNNs and diffusion models, in close collaboration with scientists and engineers.

JAX · PyTorch · XLA +6

• London
• Hybrid · 3 days in office
• Senior

Translate the company's security strategy into engineering reality, steering a multi-quarter InfoSec roadmap across Cloud Security, Platform IAM and Secure SDLC. Partner directly with engineering and ML teams to secure drug discovery pipelines and Agentic AI workflows, and build data-driven dashboards giving the CISO visibility into program health and risk reduction.

GCP · ISO 27001 · ISO 42001 +6

• London
• Hybrid
• Senior
• Permanent

Builds and evolves the data pipeline and orchestration platform behind Wayve's autonomous driving data, revamping the existing Harness system and creating a new DAG-based workflow platform. Works on Kubernetes-deployed, distributed infrastructure handling hundreds of parallel pipeline steps, batch inference and evaluation workflows, closely with partner teams.

Harness · Airflow · Dagster +7

• London
• Hybrid
• Lead
• Permanent

Owns the autonomy performance measurement framework for a public-scale robotaxi deployment, defining metrics, dashboards and release gates for an end-to-end learned driving system with no existing playbook. Runs the detect-triage-root-cause-fix loop across model, robot software, backend and evaluation tooling, leading cross-functional teams without direct authority while staying hands-on in Python, C++ and SQL.

Databricks · Looker · Grafana +2

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

Lead and contribute to core machine learning research applying large language models to drug discovery, developing and refining LLM-driven approaches across biology and chemistry problems. Work spans post-training, reasoning, test-time scaling, tool-use, alignment, agents and reinforcement learning, with model training and inference engineering in JAX, PyTorch or TensorFlow. Depending on experience, you may also lead standalone projects, mentor ML researchers and shape the research roadmap.

JAX · PyTorch · TensorFlow +3

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

Builds greenfield machine learning models and algorithms for drug discovery, designing new architectures, running experiments and scaling training and inference frameworks. Works in an interdisciplinary team of ML scientists, engineers and domain experts, with opportunities to lead projects and mentor others depending on experience.

JAX · PyTorch · TensorFlow +3

• London
• Hybrid · 3 days in office
• Senior

Lead the hardening and scaling of an inference platform serving fleets of cutting-edge ML models for scientific applications. Set strategy and roadmaps, architect next-generation services, and solve core scaling limits for high-throughput performance, partnering with research, product and operations teams.

Kubernetes · Google Cloud Platform · NVIDIA GPU +1

• London
• Hybrid · 3 days in office

A general expression-of-interest pipeline for experienced software engineers, ML researchers and engineers, computational chemists and technical program managers. The interdisciplinary team builds predictive and generative AI models for drug design, working across therapeutic areas and modalities. Hybrid working requires three days a week in the London office.

AlphaFold

• Cambridge, United States
• Hybrid · 3 days in office
• Senior · manages people

Lead translational biology for small-molecule drug discovery, designing in vitro/in vivo pharmacology, PK/PD and safety packages and biomarker strategies that bridge ML predictions to clinical translation. Oversee CRO networks and provide scientific input to ML, DMPK, toxicology and clinical teams, with some roles line-managing 2-3 in vivo pharmacologists.

Machine Learning · AlphaFold

• London
• Hybrid
• Senior
• Permanent

Lead large-scale data enrichment and curation for the Core Model Safety team, collaborating across pre/mid-training, reward modelling and self-supervised labelling. Build workflows for hard and rare event mining, design out-of-distribution detection for driverless operation, and improve ML operations efficiency across model engineering, infrastructure and compute.

PyTorch · CUDA · VLM +1

• London
• Hybrid
• Senior
• Permanent

Builds and maintains core C++/Python components that interface self-driving models with vehicle and embedded systems, plus infrastructure to run and test autonomy capabilities. Owns reliability and performance of Linux-based systems and specialises in one domain such as localisation/mapping, filtering/diagnostics or control/inference, in a team scaling to 5-6 engineers.

Linux

• London
• Hybrid
• Mid-level
• Permanent

Own a substantial work package end-to-end, defining the problem, setting strategy and sequencing work to ship outcomes for an end-to-end AI driving platform. Lead a cross-functional team of engineers, researchers and PMs without formal authority, translating R&D progress and customer needs into a product roadmap with clear KPIs and milestones.

• London
• Hybrid
• Lead
• Permanent

Cross-functional individual contributor defining software architecture across onboard software, AI, simulation, training, inference and Robotaxi systems. Leads large architectural changes spanning multiple engineering organisations, setting boundaries and interfaces between AI, compute and vehicle systems, and acts as technical tiebreaker on complex decisions.

• London
• Hybrid
• Senior
• Permanent

Set the technical direction for a petabyte-scale object storage platform underpinning autonomous driving AI, leading a multi-quarter programme from architecture through adoption. Build production services and controllers in Python or Go with Kubernetes and Terraform, optimising read performance, caching and cost while partnering with ML, data, security and privacy teams.

Kubernetes · Terraform · Azure Blob +3

• London
• Hybrid
• Senior
• Permanent

Join a small, high-autonomy team building Wayve's Data Transfer Hub, mission-critical infrastructure moving petabytes of sensor data per day from global partners like Uber, Stellantis and Nissan. You'll design and operate large-scale pipelines in Python and Terraform, using Flyte, Kafka and Azure, with ownership of production systems and an out-of-hours on-call rotation.

Flyte · Kafka · Azure +4