Machine Learning Jobs

• London
• Hybrid · 2–3 days in office
• Mid-level

Build the simulation side of Large Physics Models, generating multi-fidelity CFD data to train large-scale ML models. Develop solvers in differentiable or GPU-native frameworks like JAX or Warp, using commercial and open-source CFD tools, while collaborating closely with ML researchers in a research team.

JAX · NVIDIA Warp · OpenFOAM +2

• New York, United States
• US$150,000 – US$250,000 pa

Builds and deploys customer-facing applications that put physics AI models into production, working directly with customers, data scientists and ML engineers. Involves rapid prototyping, RESTful APIs in Python (FastAPI), microservices, and cloud deployment with Kubernetes, Helm, Terraform and AWS/Azure, plus quarterly on-site travel.

FastAPI · Helm · Kubernetes +3

• 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

• Wantage
• On-site
• Senior
• Permanent

Lead the design and deployment of autonomous robotic systems, owning software architecture across perception, planning and control. Work spans prototyping, simulation and field testing, integrating with hardware teams while mentoring engineers and shaping technical roadmaps. Requires depth in C++, ROS2, SLAM and real-time systems.

ROS · ROS2 · Gazebo +1

• London
• £90,000 – £110,000 pa
• Senior

Designs and optimises multi-threaded C++ pipelines for image processing, computer vision and rendering, tuning algorithms for spatial analysis, sensor alignment and scene reconstruction. Uses parallel computing and GPU acceleration to hit strict performance targets, builds cross-platform production software, influences architecture and joins a support rota. Works on a core platform being built from the ground up.

C++17 · Git · CMake +5

• Switzerland
• Remote
• Senior

Develops performance optimizations in deep learning frameworks using JAX, contributing to JAX core components and driving peak performance on NVIDIA hardware. Works with AI researchers to build future-proof models and tooling that bridges numerical computing, simulation and deep learning research. Requires C/C++ and Python skills plus deep knowledge of ML framework internals and CPU/GPU architectures.

JAX · PyTorch · TensorFlow +4

• United Kingdom
• Remote
• Senior

Develops performance optimizations in deep learning frameworks using JAX, contributing to JAX core components and driving peak performance on NVIDIA hardware. Works with AI researchers to build future-proof models and tooling that bridges numerical computing, simulation and deep learning research. Requires C/C++, Python and machine learning framework internals expertise.

JAX · PyTorch · TensorFlow +4

• London
• Senior

Lead the design and deployment of applied machine learning models for infrastructure and engineering, owning the full lifecycle from problem formulation to production iteration. Work spans optimisation, simulation and scientific ML, partnering with MLOps, software and domain experts to embed AI into a SaaS platform, with mentoring and technical leadership duties.

PyTorch · TensorFlow · scikit-learn

• London
• On-site
• £150,000 – £200,000 pa
• Mid-level
• Permanent

Develops and deploys robot manipulation policies using behaviour cloning and reinforcement learning, training and evaluating models on real humanoid hardware. Works within a specialist Manipulation team alongside simulation, teleoperation and controls engineers to improve sim-to-real transfer and robot performance.

PyTorch · JAX · OpenVLA +1

• London
• £60 ph
• Senior
• Contract

Develops and deploys computer vision models using CNNs, Vision Transformers and deep learning, translating research into production-ready software and scalable ML pipelines. Works with Python and PyTorch (TensorFlow advantageous) alongside engineers and researchers on AI for defence and high-performance applications, covering model evaluation, tuning, testing and CI/CD.

PyTorch · TensorFlow · Linux +1

• Brentford
• Hybrid
• £75,000 – £100,000 pa
• Mid-level
• Permanent

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.

• Midland, United States
• Hybrid
• £40,000 – £80,000 pa

You will translate operational requirements into system requirements, concepts of operation and architectures for autonomous and uncrewed systems, then support integration, verification, trials and testing. The role spans modelling and simulation, algorithm evaluation for guidance, navigation, control and perception, plus client engagement and technical proposals within multidisciplinary teams.

MATLAB · Simulink · ROS 2 +4

• London
• Remote
• £200,000 – £300,000 pa
• Lead

Own machine learning projects end to end, from experimentation through to production, building reliable, scalable systems with Python and large language models. Provide technical leadership to ML engineers, set engineering standards, lead code reviews and make architecture decisions, working closely with research and application engineering teams to improve model performance in production.

LLMs