Machine Learning Jobs

• Newcastle upon Tyne
• £25,000 – £45,000 pa
• Entry level

A self-paced traineeship taking complete beginners through AI fundamentals, Python, data handling, machine learning and Git/GitHub, plus Microsoft AI-900 certification and two tutor-assigned industry projects. On completion, candidates are placed into a Junior AI Engineer role with a UK employer within 20 miles of their location.

Microsoft Azure · Git · GitHub

• Hastings
• £25,000 – £45,000 pa
• Entry level

A self-paced traineeship taking complete beginners through AI fundamentals, Python, data handling, machine learning and Git/GitHub, with Microsoft AI-900 certification and two tutor-assigned industry projects. On completion, candidates are placed into a Junior AI Engineer role with a partner employer within 20 miles of their location.

Azure · Git · GitHub +1

• 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

• Bristol
• On-site
• Entry level
• Internship

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.

CUDA · OpenACC · OpenMP +4

• London
• Mid-level
• Permanent

Hands-on analysis and modelling of large multimodal datasets to measure and improve Rufus, Amazon's conversational shopping assistant. Designs and analyses A/B tests, builds predictive models and metrics dashboards, and works with NLP, information retrieval and recommender systems alongside scientists and engineers in a London team of around 150.

Large Language Models · Knowledge Graphs · ETL

• London
• Senior
• Permanent

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.

STARsolo · Alevin-fry · Chromap +7

• London
• Hybrid · 3 days in office
• Senior
• Permanent

Builds and operates production ML systems for document classification and extraction, using LLMs/VLMs and traditional ML to process government identity documents at scale. Owns features from RFC through deployment, optimising GPU inference and training/evaluation pipelines in a Python, AWS and Kubernetes stack, while mentoring engineers and leading technical design.

TensorFlow · PyTorch · Triton +5

• London
• Hybrid · 3 days in office
• Lead

Builds and evolves a large-scale ML infrastructure platform that supports training and deploying novel models across vast compute and data pools. Works closely with engineering, research and product teams, and explores deploying agents to improve platform robustness. Requires strong Python and experience scaling complex systems, with ML or Kubernetes exposure a plus.

Kubernetes · Google Cloud Platform · GCP

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

Join the model performance and scaling team to implement and optimize LLM post-training methods on frontier models, working with research teams to turn new methods into production systems. The role focuses on distributed training and inference performance, low-precision methods, and frameworks like JAX or PyTorch, in a hybrid London setting.

JAX · PyTorch · NCCL +5

• New York, United States
• Hybrid
• US$200,000 – US$300,000 pa
• Lead

Partners with engineering teams to embed security into CI/CD pipelines, lead threat modeling and secure design reviews for web apps, APIs and cloud services, and run the product vulnerability lifecycle. Also drives secure coding standards, delivers hands-on training, and engages with customers during security reviews.

GitHub · GitLab · Semgrep +3

• Lisbon, Portugal
• Hybrid · 3 days in office
• Senior
• Permanent

Builds and operates production ML systems for document classification and extraction, covering training, evaluation and deployment pipelines for LLM/VLM models, GPU inference optimisation and advanced labelling workflows. Leads technical design from RFC to production, mentors engineers and works across Applied Science, Product and Data Science in a squad-based team.

TensorFlow · PyTorch · Triton +4

• Singapore
• Hybrid
• Senior

Build physical foundation models for industrial engineering, scaling training across multi-GPU cloud compute and turning research prototypes into robust, optimised implementations. Work at the intersection of data science and software engineering, owning research work-streams and translating results into reusable libraries, tooling and products alongside scientists, customers and partners.

PyTorch · JAX · NumPy +7

• London
• Hybrid · 1 day in office
• £255,000 – £325,000 pa
• Senior

Builds and operates security for large-scale AI clusters across multiple clouds and bare-metal data centers, covering IAM, network segmentation, encryption, default-deny ingress/egress and mTLS with service identity. Ships controls as secure defaults via infrastructure as code and GitOps, and designs secure CI/CD and Kubernetes workflows. Also threat models multi-cloud environments, mentors engineers and supports hiring.

Kubernetes · GitOps · VPC +4

• London
• Hybrid
• Mid-level
• Permanent

Design and build production systems that evaluate AI agent behaviour in real time, detecting risky actions and enabling safe recovery or escalation. Work spans architecture, implementation, evaluation and operations, partnering closely with alignment researchers to turn prototypes into scalable safety capabilities across OpenAI products.

OpenAI

• London
• Hybrid · 1 day in office
• £325,000 – £485,000 pa
• Lead

Sets technical direction for agent-driven compute cluster lifecycle management, covering provisioning, updates and decommissioning across major clouds and own datacenters. Partners with security, research and cloud providers on secure-by-default, high-bandwidth, fault-tolerant clusters, and mentors engineers while owning multi-quarter strategy.

Kubernetes · Terraform · AWS +7