Machine Learning Jobs

• Singapore
• Senior

Build and deploy customer-facing applications that put physics AI models into production, working directly with customers, data scientists and ML engineers. The role spans rapid prototyping, backend development of Python/FastAPI REST APIs and microservices, and cloud deployment using Kubernetes, Helm, Terraform and AWS/Azure. It suits someone who can grasp complex engineering problems across industries and own solutions end-to-end.

FastAPI · Helm · Kubernetes +3

• London
• Hybrid
• Lead
• Permanent

Lead large-scale training and post-training for Gaia, Wayve's world model that predicts future driving video frames and runs closed-loop with the driving model. Own model architecture and training-strategy decisions, push autoregressive generation toward longer, more stable rollouts, and make the model deployment-ready. Work across Research, Applications, Core Simulation Engineering and Cloud/Infrastructure, mentoring and setting technical direction.

PyTorch

• London
• Hybrid
• Director
• Permanent

Owns the architecture, strategy and integrity of safety cases for robotaxi and OEM ADAS products, defining top-level claims, argumentation structures and evidence traceability. Sits independently from Engineering and Product, providing expert challenge and oversight across AI/ML, systems, validation and operational safety. Focuses on safety arguments for AI-enabled autonomous driving, including functional safety, SOTIF and post-deployment monitoring.

ISO 26262 · ISO 21448 · ISO/PAS 8800 +2

• Switzerland
• Remote
• Senior
• Permanent

Builds production-quality C++ components, APIs and integrations for NVIDIA's AI-powered developer-tools ecosystem, connecting native C++ tools to retrieval and agentic services via Python, MCP and REST. Involves architecting performance-sensitive systems, contributing to agentic workflows and mentoring engineers.

MCP · REST APIs · Python +2

• 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

• Lausanne, Switzerland
• Hybrid · 3 days in office
• Senior
• Permanent

Architect and secure an AI-first drug discovery platform, reporting to the CISO. You will build AI/ML threat models, protect model artifacts, deploy LLM and agentic guardrails, and lead AI-specific incident response across distributed training and inference infrastructure.

JAX · PyTorch · TensorFlow +7

• London
• Hybrid
• £65,000 – £145,000 pa

The Chem Bio team researches, evaluates and mitigates chemical and biological risks from frontier AI systems, running capability evaluations, building benchmarks and datasets, and developing safeguards and monitoring methods. Findings are shared with frontier AI companies, UK officials and allied governments to inform deployment and policy. The role is open to research engineers and scientists at all seniority levels, with appointment conditional on obtaining UK Government Developed Vetting clearance.

• London
• Hybrid · 3 days in office
• Senior

Senior individual contributor building the shared platform, tooling and infrastructure for an interdisciplinary chem-bio research team, including LLM-based agent systems and evaluation frameworks. Works closely with researchers to translate varied needs into maintainable Python systems, while contributing to engineering standards and mentoring peers without line management.

AWS · Kubernetes · Slurm +2

• Cambridge
• £63,606 – £67,468 pa
• Senior
• Permanent

Contribute to research and teaching in artificial intelligence and machine learning as a faculty member, developing novel models, abstractions and architectures. The role involves supervising Masters and PhD students, creating graduate content for the MPhil in Advanced Computer Science, and teaching undergraduate machine learning and mathematics. The appointee is expected to secure research funding and build a world-class team, collaborating across computer science subdisciplines and with fields such as mathematics, engineering and physics.

• London
• Hybrid
• £225,000 – £255,000 pa
• Mid-level

Join the Beneficial Deployments Life Sciences team to help labs automate experiments end-to-end with Claude, connecting software to robots, instruments and LIMS. You'll prototype and productionise LLM-powered tools, build evals for physical-world safety, and scale learnings through cookbooks and the Model Hardware Standard. Expect significant travel and EMEA ambassadorship at conferences.

Claude · Model Hardware Standard · Liquid Handlers +2

• Eastleigh
• Hybrid
• £50,000 – £70,000 pa
• Permanent

Builds and supports data products on a modern cloud data platform, developing and optimising pipelines and integrating varied sources with strong quality standards. Works in agile squads with data and AI engineers, scientists and architects, using PySpark, Python, SQL and Databricks or Snowflake on AWS, plus Airflow, Terraform and Kubernetes.

Databricks · Snowflake · AWS +7

• London
• Lead · manages people
• Permanent

Lead the science vision for Trusted Supply, safeguarding advertiser trust across Amazon Advertising by developing algorithms for brand safety, suitability scoring, fraud detection, viewability prediction and privacy-preserving measurement. The role spans multi-year research direction, publication roadmaps, cross-team modelling frameworks and production systems serving billions of ad impressions daily, alongside hiring and mentoring applied scientists and research engineers.

• United Kingdom
• Remote
• Lead

Customer-facing technical authority for Generative AI across EMEA digital-native and startup accounts, shaping architectures and de-risking LLM use cases from RAG and agentic systems to fine-tuning and evaluation. Builds hands-on proofs of concept on Databricks (Mosaic AI, MLflow, Model Serving, Vector Search, Unity Catalog) and mentors a ~70-person field engineering organisation.

Databricks · Mosaic AI · MLflow +7

• 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

• Sydney, Australia
• Hybrid
• Senior
• Permanent

Lead the design, build and deployment of production-grade ML systems for diverse clients, covering architecture, scalable infrastructure and ethical standards. Work across engineering, data science and commercial teams, acting as a trusted technical advisor while mentoring junior engineers. Requires deep Python, TensorFlow/PyTorch, cloud and Kubernetes expertise.

TensorFlow · PyTorch · AWS +5