Senior AI Engineer

JR United Kingdom; Livingston; Hybrid; GBP 60,000 - 85,000; Full time

Job summary

A leading company in legal technology is seeking a Senior AI Engineer to redefine how legal teams work by building innovative AI-native experiences. This role involves developing backend systems, optimizing LLM performance, and collaborating across teams to deliver impactful functionalities in a hybrid work environment.

Benefits

Competitive salary

Equity opportunities

High-trust, high-impact environment

Qualifications

• 4+ years of backend or AI-focused engineering experience.
• Experience with LLMs and frameworks (LangChain, LangGraph).
• Strong programming skills, preferably in Python.

Responsibilities

• Build and maintain core backend systems and APIs for LLM features.
• Design and implement evaluation pipelines for model performance.
• Collaborate with legal engineers to build compliant AI systems.

Skills

Python

Backend Development

Machine Learning

API Design

Data Privacy

Tools

LangChain

PyTorch

Job description

Social network you want to login/join with:

We’ve just secured a whopping Series A—come help us shape what’s next.

Location: Edinburgh (Hybrid: 3 days in-office, 2 remote)

Mission:

Join us in redefining how legal teams work by building state-of-the-art AI-native experiences powered by Large Language Models. As a Senior AI Engineer, you’ll work across backend systems, evaluation frameworks, and agentic LLM workflows to create scalable, secure, and continuously improving AI features. You’ll partner closely with design, product, and legal teams to bring thoughtful, high-impact functionality to life.

Key Responsibilities:

• Build and maintain core backend systems and APIs powering LLM-driven features
• Design and implement evaluation pipelines to assess and optimize model performance
• Prototype and productionize agentic workflows that leverage tool usage and chaining
• Experiment with, tune, and benchmark different LLMs (e.g., OpenAI, HuggingFace, LangChain)
• Automate model evaluation and integrate insights into the dev lifecycle
• Support semantic search, data extraction, and document analysis use cases
• Collaborate with legal engineers and product to build secure and compliant AI systems
• Contribute to architectural decisions, internal tooling, and team best practices

Experience & Qualifications:

• 4+ years of backend or AI-focused engineering experience
• Hands-on experience working with LLMs and related frameworks (LangChain, LangGraph, etc.)
• Strong programming skills—Python preferred; experience with APIs and cloud infra is a plus
• Proven track record designing evaluation frameworks and tuning model performance
• Solid understanding of data privacy, system integrity, and scalable architecture
• Experience with semantic search, retrieval pipelines, or document-based AI systems is a bonus

Nice to Have:

• Exposure to legal tech, compliance tools, or privacy-centric workflows
• Familiarity with ML frameworks (e.g. PyTorch, Scikit-learn)—helpful but not required

What We Offer:

• A high-trust, high-impact environment with ownership over mission-critical AI systems
• Challenging product problems at the intersection of law, language, and intelligence
• A collaborative team of engineers, designers, and legal experts building something new
• Competitive salary, meaningful equity, and a say in shaping both product and culture

Who You Are:

You’ve worked at the cutting edge of AI or backend systems and are excited about applying that experience to real-world legal challenges. You’re hands-on, evaluation-obsessed, and motivated by tangible user outcomes. You thrive in fast-moving environments and enjoy defining best practices as you go.