Machine Learning Jobs

16 jobs for “Graph Representation Learning”

Results

Hadean logo

Hadean · 11–50 people · Series A

Engineering Manager - AI Defence Product (dominAI)

  • London
  • Hybrid · 1 day in office
  • Senior · manages people
  • Permanent

Lead a multidisciplinary engineering team building dominAI, an AI-powered defence product, line-managing AI/ML, simulation, infrastructure and web engineers while staying hands-on in TypeScript, React and Node. Own delivery end-to-end, hiring and people development, partnering with Product and technical leads. Hybrid, one day a week in Shoreditch.

TypeScript · React · Node

Featured SC clearance 2 weeks ago
CureAge Therapeutics logo

CureAge Therapeutics · 1–10 people · Seed

Senior Computational Chemist, LNP Design

  • London
  • Hybrid
  • Senior
  • Permanent · Part time

Lead the computationally guided design of a lipid nanoparticle delivery platform, applying chemical and biophysical principles to optimise LNP composition and formulation. Use AI-enabled chemistry, cheminformatics, predictive models and molecular modelling to prioritise formulation components, working closely with the experimental team and founding team in a small start-up.

Featured 3 weeks ago

Wayve · 11–50 people · Series A

Principle Engineer, 3D Reconstruction

  • London
  • Hybrid
  • Lead
  • Permanent

Lead the technical direction for an offline 3D reconstruction system that turns raw vehicle sensor data into quality-assessed 3D ground-truth geometry and ego motion. The role spans SLAM, lidar point-cloud registration, state estimation and sensor fusion, with responsibility for architecture, roadmaps, interfaces and cross-team alignment. It is a principal-level leadership position mentoring engineers and setting reconstruction quality standards.

ROS · PCL · OpenCV +3

Visa sponsorship Last month

NES Group Ltd

Senior Data Scientist

  • London
  • Senior
  • Contract

Builds ML and graph-based identity resolution where rule-based matching falls short, improving customer-spine architecture and reducing identity drift. Defines matching-quality metrics, supports golden-layer integration and delivers a reusable identity-resolution framework. Requires entity resolution, fuzzy matching, MLOps and feature-store skills plus Customer Master/MDM experience.

MDM

2 weeks ago

NVIDIA · Public

Compiler Engineer - AI Inference

  • Cambridge
  • Mid-level

Hands-on compiler engineering focused on kernel generation and computational graph optimizations for next-generation NVIDIA GPUs, solving complex compilation problems for AI inference and training workloads. Works with MLIR, C/C++ and Python, collaborating with software, hardware and research teams on hardware/software co-design and datacenter-scale AI deployment.

MLIR · GPU · CPU +1

3 months ago

NVIDIA · Public

Compiler Engineer - AI Inference

  • United Kingdom
  • Remote

Hands-on development of kernel generation and computational graph optimizations for next-generation NVIDIA GPUs, solving complex compilation problems for AI inference and training workloads. The role involves hardware/software co-design with software, hardware and research teams, and scaling datacenter AI deployments, with heavy use of MLIR, C/C++ and Python.

MLIR · GPU · CPU +1

3 months ago

PhysicsX

Machine Learning Engineer

  • San Francisco, United States
  • Hybrid
  • US$150,000 – US$190,000 pa
  • Mid-level

Builds and deploys machine learning pipelines for industrial physics applications, working directly with customers on-site to embed AI models into practical tools. Involves manipulating 3D point-cloud and mesh data, cloud and HPC infrastructure, and translating R&D outputs into reusable libraries and products.

TensorFlow · MLflow · Spark +6

2 days ago

Get new Graph Representation Learning roles by email. One email a week when new roles match, or daily if you prefer. Stop any time.

PhysicsX

Machine Learning Engineer

  • New York, United States
  • Hybrid
  • US$150,000 – US$190,000 pa
  • Mid-level

Builds and deploys machine learning systems for industrial physics applications, working directly with customers on-site to embed AI models into practical tools. Involves designing scalable data pipelines, manipulating 3D point-cloud and mesh data, and translating R&D outputs into reusable libraries and products. Requires travel to customer sites across multiple continents for 3-4 weeks per quarter.

TensorFlow · MLflow · Spark +6

2 days ago
Healx logo

Healx · 51–200 people · Series B

Software Engineer (Agentic Systems)

  • Cambridge
  • Hybrid
  • Mid-level

Builds and refines LLM-agentic workflows on an internal framework to help drug discovery scientists generate and triage testable therapeutic hypotheses for rare diseases. Integrates agents with a knowledge graph, proprietary methods and scientific literature, and contributes to evaluation, testing and quality standards. Works in a cross-functional team of ML engineers, bioinformaticians, scientists and clinicians, with scope to own workflows end to end.

Python · ADK · AgentSpace +1

3 months ago

University of Exeter

PhD Studentship: Next-generation machine learning for genetic interaction prediction with lab-in the-loop validation in pancreatic cancer

  • Exeter
  • £21,805 pa
  • Entry level
  • Contract

Builds next-generation machine learning models to predict genetic interactions from large-scale perturb-seq data, evaluating graph neural networks and autoencoders for genome-wide scalability and cross-cancer generalisation. Works in a lab-in-the-loop cycle with AstraZeneca and Cancer Research Horizons FGC, who generate pancreatic cancer perturb-seq data to fine-tune the computational models. Purely computational, with high performance computing and industry-aligned experience.

Perturb-seq · DepMap · Project Score

2 weeks ago

HealthHero

Senior AI Engineer

  • London
  • Senior

Owns end-to-end delivery of AI initiatives, starting with a Class II medical device system combining an AI layer with deterministic components. Involves NLP, LLMs, knowledge graphs, evaluation pipelines and clinician review workflows, working closely with clinical, quality and product stakeholders in a regulated environment.

AWS · GCP · Azure +7

4 days ago

HealthHero

Senior AI Engineer

  • London
  • Senior

Own the end-to-end technical delivery of AI initiatives, starting with the Clinical Brain, a Class II medical device combining an AI layer with deterministic components. Work with clinical, quality and data teams on NLP, LLMs, knowledge graphs, evaluation pipelines and LLMOps, building to regulated safety and traceability standards.

Neo4j · Cypher · GraphRAG +6

4 days ago
Isomorphic Labs logo

Isomorphic Labs · 201–500 people · Series A

Research Scientist

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

Builds greenfield machine learning models and algorithms for drug discovery, designing architectures, running experiments and scaling training and inference frameworks. Works in an interdisciplinary team of ML scientists, engineers and domain experts, with opportunities to lead projects and mentor others. Hybrid role requiring three days a week in the London office.

JAX · PyTorch · TensorFlow +3

2 days ago
Isomorphic Labs logo

Isomorphic Labs · 201–500 people · Series A

Research Scientist (Applied LLMs), London

  • London
  • Hybrid · 3 days in office
  • Senior · manages people

Lead and contribute to core machine learning research applying large language models to drug discovery, developing and refining LLM-driven approaches across biology and chemistry problems. Work spans post-training, reasoning, test-time scaling, tool-use, alignment, agents and reinforcement learning, with model training and inference engineering in JAX, PyTorch or TensorFlow. Depending on experience, you may also lead standalone projects, mentor ML researchers and shape the research roadmap.

JAX · PyTorch · TensorFlow +3

4 months ago

Wayve · 11–50 people · Series A

Senior ML Performance Engineer, Inference Optimisation

  • London
  • Hybrid
  • Senior
  • Permanent

Hands-on optimisation of large transformer model inference for edge accelerators and GPUs, profiling model graphs, compiler/runtime behaviour, kernel execution and memory movement. You'll implement fusions, scheduling, quantisation-aware performance and custom kernels, and build benchmarking and regression tests for targets like NVIDIA Orin/Thor and Qualcomm platforms, working closely with model, platform and deployment teams.

TensorRT · CUDA · Qualcomm QNN +6

Visa sponsorship Last month

Cambridge Consultants

Graduate Human-Centric AI Engineer (2027 start)

  • United Kingdom
  • Entry level

Graduate role in the Human-Machine Understanding group, working at the intersection of AI, cognitive science and human-machine interaction. You will contribute to an Intelligent Interfaces framework spanning multi-modal human sensing, human-centric contextual modelling and agentic workflows, supporting client projects from human subjects studies to technical prototypes. Requires Python and AI/ML frameworks such as PyTorch, with scope to explore HRI, affective computing and cognitive architectures.

PyTorch · ROS2

SC clearance Last month

Filters

Work policy

Job type

Seniority

Are you sure?