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

A paid 6-12 month PhD internship in the AI & Algorithms department, working across the full robotic learning pipeline from data collection and model training to deployment on real hardware. Interns join mixed-discipline teams on client projects in healthcare and industrial settings, contributing to internal R&D and using labs, robotic platforms and compute resources.

Cambridge Consultants · United Kingdom

Join the AI & Algorithms department to build intelligent systems, coding hands-on and training/testing models for client projects across robotics, radar, photonics and engineering biology. Work in mixed-discipline teams attending stand-ups and planning sessions, while contributing to internal R&D that turns state-of-the-art literature into market-leading solutions and client demonstrations.

NVIDIA · Bristol

Research and develop techniques to GPU-accelerate deep learning, machine learning and other AI workloads, working with technical experts from industry and academia to analyse and optimise complex AI and HPC algorithms on CPU and GPU architectures. The role also involves publishing optimisation techniques via developer blogs or conferences and influencing next-generation hardware, software and programming models.

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.

Isomorphic Labs · London

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.

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.

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.

Amazon · London

PhD students join an Amazon Science team to own an end-to-end research project, from scoping with a mentor to deploying algorithms, models or systems into production. Projects span computer vision for Prime Video quality, multimodal AI for Rufus, automated reasoning and formal verification, observability for distributed systems, and recommendation models. Interns may publish at top-tier conferences or see their work ship before the internship ends.

Relation Therapeutics · London

Builds and scales generative and predictive models of cellular behaviour, acting as the bridge between ML theory and implementation. Day to day involves optimising large neural networks, distributed multi-GPU/multi-node training, profiling GPU workloads, and productionising research code for multi-modal biological data.