Machine Learning Jobs

NVIDIA · United Kingdom

Develop algorithms and optimizations for NVIDIA's LPX inference and compiler stack, working at the intersection of large-scale systems, compilers, and deep learning. The role involves building high-performance runtime and compiler components, mapping inference workloads onto future hardware, and collaborating with hardware architects on codesign. Distinct focus on LLVM/MLIR, graph transformations, and spatial accelerators.

NVIDIA · Zürich, Switzerland

Design, develop and optimise GPU/CPU-accelerated data processing libraries such as DALI and nvImageCodec, covering numerical methods, performance tuning and benchmarking across architectures. Works closely with leadership and external partners on technical roadmaps while providing technical leadership and guidance to other library engineers.

Wayve · London

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.

Cambridge Consultants · United Kingdom

Work as part of a multi-disciplinary technical team on client assignments spanning consumer devices, industrial systems, and defence. Design and implement embedded systems, including C software for embedded targets, SystemVerilog for FPGAs, PCB design, and hardware testing, while interacting directly with clients from requirements capture to handover.

Relation Therapeutics · London

Processes and QC's paired single-cell RNA-seq and scATAC-seq at scale, building Nextflow pipelines and multimodal training datasets in Zarr for GPU-accelerated model training. Partners with wet lab scientists and ML modellers, contributing to evaluation frameworks that test whether models capture meaningful biology.

Anthropic · London · £370,000 – £630,000 pa

Builds and improves the core RL training infrastructure that researchers depend on, profiling, debugging and rearchitecting bottlenecks across the stack. Works closely with researchers and adjacent teams such as inference and sandboxing to ship tooling that speeds up training runs, spanning low-level performance work through to RL algorithms.

Anthropic · London · £260,000 – £630,000 pa

Join the Pretraining team to develop next-generation large language models, working across model architecture, algorithms, data processing and optimizer development. You'll lead small research projects, design and analyse experiments, and scale distributed training infrastructure to thousands of GPUs, balancing research goals with practical engineering.

Isomorphic Labs · Cambridge, United States

Lead medicinal chemistry efforts on drug discovery projects, guiding them from initiation to clinical candidate selection as a Project Lead or Medicinal Chemistry Lead. Partner closely with ML scientists to refine AI models, manage CRO resources, and generate IP, with some roles line-managing 2-4 chemists.

Isomorphic Labs · Cambridge, United States

Lead computational drug design programs from hit identification through to candidate nomination, applying AI-first approaches and shaping the ML drug design engine alongside ML scientists. Roles span Program Lead or Computational Chemistry Lead, with some positions managing 2-3 computational chemists. Hybrid working in Cambridge, MA, three days a week in office.

Amazon · Cambridge

Joins a research team advancing multimodal deep learning for consumer devices, focusing on dataset curation, annotation pipelines and automated evaluation methods such as fine-tuned LLM-as-judge. Analyses large offline and online datasets in Python to diagnose model failures and works with training, compiler and product teams to move models from prototype to production.

Databricks · London

Owns the core write-path operations for Delta Lake, the open-source storage layer behind the Databricks Lakehouse, driving performance innovations like Low Shuffle Merge and Deletion Vectors. Involves low-level systems debugging, performance optimisation on large production clusters, architecture design, and technical leadership including mentoring and sprint planning.

Faculty · London

Lead high-impact AI projects for retail and consumer clients, defining data science approaches, designing software architectures, and mentoring colleagues. Own the technical side of client relationships, scope projects, and embed responsible AI practices while delivering bespoke algorithms and production-grade systems.

MFK Recruitment · Brentford · £75,000 – £100,000 pa

Develop an engineering-grade spiking neural network engine for machine vision applications in defence and security contexts. Translate advanced research into robust production software using C++ and machine vision techniques. Collaborate with researchers and engineers on SNNs, CNNs, and Transformers, with integration into operational systems.

Ocado · London

Owns the software engineering that brings data science and machine learning products into live production for the Autoplanner team, building AI-powered planning systems across global warehouses and last-mile assets. Writes typed, well-tested Python (with some Java for OSP integration), builds APIs and data contracts, manages AWS deployments and CI/CD, and packages, serves and monitors ML models. Acts as engineering partner to data scientists and shares on-call support.

Cohere · London

Hands-on research and engineering role on the Agent Code team, developing code-generating LLMs and autonomous agent systems for enterprise use. Involves training code models, building agent frameworks for inference and sampling, creating SFT trajectories, working on RL algorithms, and designing benchmarks, in collaboration with the pretraining team on frontier compute infrastructure.