Artificial Intelligence Jobs

• London
• Hybrid · 3 days in office
• Senior

Owns deployment of ML models and engineering surrogates (deep learning on CAE/CFD/FEA data, forecasting, anomaly detection) into customer production environments, building scalable training and inference pipelines with Python, PyTorch, Kubeflow and fastAPI. Leads scoping and architecture, mentors engineers, and travels to customer sites roughly 3-4 weeks per quarter.

Python · PyTorch · Pandas +6

• Sydney, Australia
• Hybrid
• Senior
• Permanent

Lead the design, build and deployment of production-grade ML systems for a range of clients, covering architecture, scalable infrastructure and engineering standards. Work across the full ML lifecycle with TensorFlow or PyTorch, cloud platforms and Docker/Kubernetes, while advising customers and mentoring junior engineers.

TensorFlow · PyTorch · AWS +5

• Singapore
• Lead

Lead the deployment of ML models and engineering surrogates—deep learning on CAE/CFD/FEA data, time-series forecasting, anomaly detection, and optimisation—into customer production environments. Own scoping, architecture, and success metrics for data/ML systems built with Python, PyTorch, Pandas, fastAPI, Scipy, and Kubeflow across cloud and on-prem. Mentor engineers and data scientists while travelling 3-4 weeks per quarter to customer sites worldwide.

Python · PyTorch · Pandas +3

• London
• Hybrid
• Senior
• Permanent

Lead the design, build and deployment of production-grade ML systems for national security and AI safety clients, including capability testing of frontier models. Act as a technical advisor to customers, define best practices for deploying ML at scale, and mentor junior engineers. Distinct for its focus on red teaming, evaluation and high-stakes government and frontier lab work.

TensorFlow · PyTorch · AWS +4

• 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 and builds CI/CD tooling and observability to deliver ML systems reliably at scale.

PyTorch · TensorRT · GitHub Actions

• London
• Hybrid
• Lead
• Permanent

Lead large-scale training and post-training for Gaia, Wayve's world model that predicts future driving video and runs closed-loop with the driving stack. Own model architecture decisions, push autoregressive rollouts toward longer, more stable generation, and make the model deployment-ready. Partner with research, applications, simulation engineering and cloud teams, mentoring and setting technical direction.

PyTorch

• Brentford
• Hybrid
• £70,000 – £95,000 pa
• Senior
• Permanent

Design and deploy production-grade computer vision systems using deep learning techniques such as CNNs and Vision Transformers, translating research into robust AI solutions for defence and security applications. Work across the full development lifecycle, from model design and training to deployment on cloud and edge platforms. Collaborate with researchers, engineers, and customers in a technically challenging environment focused on real-world autonomy and perception systems.

• Cambridge
• Hybrid
• Senior

Develops algorithms and optimizations for NVIDIA's LPX inference and compiler stack, building high-performance runtime and compiler components for end-to-end inference. Involves mapping large-scale neural network workloads onto future hardware, working with LLVM/MLIR, and collaborating with hardware architects on codesign. Hybrid role based in Cambridge.

LLVM · MLIR · TensorFlow +3

• United Kingdom
• Hybrid
• Senior

Develops algorithms and optimizations for NVIDIA's LPX inference and compiler stack, building high-performance runtime and compiler components and mapping large-scale inference workloads onto future hardware. Works at the intersection of compilers, systems and deep learning, using LLVM/MLIR and frameworks like PyTorch and TensorFlow, and collaborates with hardware architects on codesign.

LLVM · MLIR · TensorFlow +3

• London
• Hybrid
• Senior

Owns the software engineering that brings data science and ML products into live production for the Autoplanner team, building AI-powered planning systems for 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 duties.

AWS · GCP · Ocado Smart Platform

• London
• Hybrid · 1 day in office
• £260,000 – £630,000 pa
• Mid-level

Blends reinforcement learning research with engineering, building and scaling RL infrastructure, training environments and evaluations for large language models. Work spans agentic tool use, code generation, reasoning and distributed GPU training, with close collaboration across research, alignment and production teams.

PyTorch · TensorFlow · JAX +3

• London
• Hybrid
• £370,000 – £630,000 pa
• Mid-level

Builds and improves the RL training infrastructure researchers rely on, profiling and rearchitecting bottlenecks across the stack and owning run reliability end-to-end. Works closely with researchers and adjacent teams like inference and sandboxing, spanning low-level performance work through to RL algorithms.

JAX · PyTorch

• London
• Hybrid
• Senior
• Permanent

Architect novel generative models to design functional proteins, working in an interdisciplinary team of ML researchers, protein engineers and biologists. The role spans data curation, evaluation metrics, rapid prototyping, cloud training and inference, and planning wet lab testing campaigns, with about 10% of time for self-development.

• Huddersfield
• Hybrid
• £55,000 – £70,000 pa
• Mid-level

Develops software and applied AI models for machine vision, working across C++, C# and Python to build image-processing algorithms and integrate models into reliable applications. Involves dataset preparation, model evaluation, performance tuning for accuracy and inference speed, automated testing and debugging of vision pipelines. Hybrid or remote, with a path to lead an AI team.

OpenCV · PyTorch · TensorFlow +4

• London
• Hybrid · 3 days in office
• Lead

Technically lead the Platform group, designing, operating and refining cloud infrastructure across research, development and production. Work spans GCP, Kubernetes, Docker, Terraform and Kapitan, plus CI/CD, observability and tooling, partnering with scientists, ML researchers and product teams. Reports to the Director of Engineering in a small, fast-moving team.

GCP · Kubernetes · Docker +7