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.

Cambridge Consultants · United Kingdom

Take technical ownership of AI projects for defence and security clients, designing and building agentic systems that search, interpret and act on information. Lead smaller projects, mentor junior engineers, and work directly with clients on proposals and new work. Requires SC clearance and UK residency.

Cambridge Consultants · United Kingdom

Senior technical role architecting agentic AI systems for defence and security clients, acting as technical lead on complex projects and setting departmental technical direction. Involves hands-on engineering in Python, client-facing proposal work, mentoring engineers and conducting technical interviews. Requires SC security clearance and UK residency.

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.

Amazon · Edinburgh

Builds machine learning recommendation systems that match job seekers to roles and recruiters to talent across Amazon Jobs and internal hiring platforms. Works on representation learning, reinforcement learning and probabilistic modelling at global scale, with a focus on fairness and explainability, alongside engineers, scientists and product managers through the full development lifecycle.

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.

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.

Anthropic · London · £325,000 – £390,000 pa

Builds and operates the distributed systems that serve Claude to millions of users, spanning request routing, load balancing, autoscaling and fleet orchestration across thousands of AI accelerators in multiple clouds. The role sits at the intersection of high-performance infrastructure and ML, integrating new accelerator platforms and supporting new model architectures.

Wayve · London

Lead the technical direction for an offline 3D reconstruction system that turns raw vehicle sensor data into quality-assessed 3D ground-truth geometry and ego motion. The role spans SLAM, lidar point-cloud registration, state estimation and sensor fusion, with responsibility for architecture, roadmaps, interfaces and cross-team alignment. It is a principal-level leadership position mentoring engineers and setting reconstruction quality standards.

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.

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.

Cohere · London

Drive safety research for agentic LLMs, focusing on data generation, post-training algorithms and evaluation methods so models that access external tools and act in the world remain fair, trustworthy and secure. Work with cross-functional ML, data annotation, product and policy teams on a small, autonomous team, requiring strong statistics, software engineering and Python/PyTorch skills.