Latest Graph Representation Learning Jobs

Spotlight
Fractile logo

ML Runtime Engineer (Mid-Level and Senior)

Design and develop high-performance ML runtime systems for AI accelerators, integrating with open-source frameworks like PyTorch and vLLM. Work closely with hardware and software teams in a co-design environment to optimize inference performance. Build low-level runtime components in Rust and contribute to the full stack of ML inference infrastructure.

Fractile London, United Kingdom
Hybrid Permanent

Machine Learning Scientist

Develop novel machine learning methods for molecular and protein discovery, bridging cutting-edge research with practical engineering implementation. Work on geometric deep learning, diffusion models, and 3D representations in biological systems, translating scientific ideas into deployable tools. Collaborate closely with engineers and scientists in a hybrid London-based role focused on high-impact AI for life sciences.

Platform Recruitment London, United Kingdom £85,000 – £100,000 pa
NVIDIA logo

Relational Foundation Model Engineer, Modern Data Stack

Design and develop novel Transformer and GNN-based foundation models that generalize across relational databases and heterogeneous graphs. Work across the full ML lifecycle—from architecture design to inference optimization—on real-world applications like fraud detection, forecasting, and recommendation systems. Collaborate with researchers and engineers to build a unified AI model for enterprise data.

NVIDIA Germany
On-site Permanent
NVIDIA logo

Relational Foundation Model Engineer, Modern Data Stack

Design and develop novel Transformer and GNN-based foundation models that generalize across relational databases and heterogeneous graphs. Work across the full ML lifecycle—from architecture design to inference optimization—on a unified model powering enterprise AI applications in forecasting, fraud detection, and recommendations. Collaborate with researchers to advance relational reasoning and deploy cutting-edge systems into production.

PhysicsX logo

Machine Learning Engineer

A Machine Learning Engineer will collaborate with simulation engineers and customers to solve complex physics and engineering challenges using AI. The role involves building scalable ML pipelines, working with 3D point-cloud and mesh data, and translating R&D into deployable tools. Engineers will also travel to customer sites globally to implement solutions on-site, ensuring practical, impactful delivery across industries like aerospace, energy, and semiconductors.

PhysicsX England £150,000 – £190,000 pa
On-site Permanent Clearance Required
Healx logo

Agentic AI Engineer (life sciences)

This role involves designing and building LLM-powered agentic workflows that assist drug discovery scientists in generating testable therapeutic hypotheses for rare diseases. The engineer will work cross-functionally to integrate AI agents with a knowledge graph, scientific literature, and proprietary methods, refining systems that reason over complex biomedical data. Success includes owning workflows from concept to production, with a focus on reliable, evaluable, and maintainable agent design in a fast-moving AI landscape.

Healx Cambridge, United Kingdom
Hybrid Permanent
Wayve logo

Staff ML Performance Engineer (Compiler)

This role focuses on optimizing machine learning inference performance for edge accelerators and GPUs, particularly for running large transformer models efficiently on low-power in-vehicle systems. The engineer will work across compilers, runtimes, and kernels to deliver measurable improvements in latency, memory, and power, while collaborating closely with model developers and contributing to early-stage, high-impact projects. The position involves deep technical work on multiple hardware platforms and shaping performance engineering standards across the team.

Wayve London, United Kingdom
Wayve logo

Staff ML Performance Engineer (Inference Optimisation)

Optimise machine learning inference performance for edge and GPU accelerators, focusing on efficient execution of large transformer models in low-power, cost-constrained environments. Work across the full stack—from model graphs and compilers to kernels and embedded deployment—to deliver measurable improvements in latency, memory, and power. Collaborate with model developers and contribute to tooling and technical roadmaps for production-grade autonomous driving systems.

Wayve London, United Kingdom
Hybrid Permanent
Amazon logo

Machine Learning Engineer II, Intelligent Talent Acquisition - Lead Generation & Detection Services

This role involves building and scaling AI/ML-powered systems to transform talent acquisition at Amazon, with a focus on developing foundational data platforms and knowledge graphs. You'll work closely with scientists and engineers to deliver machine learning solutions across the full recruitment lifecycle, leveraging deep learning and generative AI. The position emphasizes software development for large-scale systems that support hiring at global scale.

Amazon Edinburgh, Alba / Scotland, United Kingdom
Hybrid Permanent
PhysicsX logo

Machine Learning Engineer

A Machine Learning Engineer will collaborate with simulation engineers, data scientists, and customers to develop AI-driven solutions for engineering and manufacturing challenges. The role involves building scalable ML pipelines, working with 3D point-cloud and mesh data, and translating R&D into deployable tools. The position requires strong software engineering practices and customer-facing delivery, with opportunities for international travel.

PhysicsX United Kingdom £150,000 – £190,000 pa
Remote Permanent

Head of Artificial Intelligence (Gen-AI)

Leads the strategic and technical development of cutting-edge Generative AI solutions, including a proprietary RAG-based conversational agent, integrated across business intelligence and compliance services. Drives end-to-end design, deployment, and optimisation of LLM pipelines and agent workflows in production, while mentoring engineering teams and shaping the AI function. Focuses on high-quality, low-hallucination outputs through advanced data-chunking, semantic search, and model evaluation.

The Portfolio Group Manchester, United Kingdom £130,000 – £150,000 pa
Hybrid Permanent Clearance Required

Data Scientist

This role involves developing and deploying data science solutions to analyze customer contact journeys using machine learning, NLP, and Generative AI. The Data Scientist will work hands-on with structured and unstructured data, build predictive models, and integrate solutions into production, particularly within Azure cloud environments. Emphasis is placed on extracting insights from text and delivering actionable outcomes using tools like SQL, Python, and PySpark.

TRIA Slough, Berkshire, United Kingdom £480 – £520 pd

Senior Engineer – AI & Workplace Enablement

This role involves modernising the digital workplace by leading the implementation and governance of Microsoft 365 E5 services, including identity, endpoint management, security, and compliance. A key focus is driving AI adoption through Microsoft 365 Copilot and Copilot Studio, building automation, and delivering training to enhance productivity. The engineer will also develop AI governance frameworks, create operational dashboards, and provide technical leadership across the platform.

dunnhumby London, United Kingdom
Hybrid Permanent

Agentic Systems Engineer (Digital Services)

Design and build proactive, human-centred agentic systems that reason, plan, and act across domains like biotech, robotics, and AI. Develop full-stack architectures from edge reasoning to cloud deployment, integrating multimodal sensing and persistent memory. Work in agile, cross-disciplinary teams to rapidly prototype and productionise breakthrough systems for high-impact clients.

Cambridge Consultants United Kingdom
Hybrid Permanent Clearance Required

Data Scientist

This role involves leading end-to-end data science projects focused on customer journey insights by combining structured and unstructured data. The Senior Data Scientist will apply NLP and GenAI techniques to extract meaning from text and interaction data, build scalable pipelines, and deliver actionable insights to stakeholders. A strong emphasis is placed on model deployment, code quality, and mentoring within an agile, collaborative environment.

Pontoon Welwyn Garden City, Hertfordshire, AL8 6TP, United Kingdom £550 – £800 pd
Hybrid Contract

Forward Deployed Engineer – AI Engineer

This role involves rapidly prototyping and delivering advanced AI solutions—spanning GenAI, RAG, agentic workflows, and safety controls—by working closely with business teams and delivery pods. The engineer bridges product, data science, and engineering to turn user needs into production-viable AI capabilities, with a strong focus on evaluation, security, efficiency, and reusable patterns. Emphasis is placed on end-to-end ownership, from discovery to handover, ensuring solutions are scalable, supportable, and aligned with enterprise standards.

PRACYVA United Kingdom £350 – £400 pd