Senior AI Engineer

Wordsmith AI Ltd; City of Edinburgh; Hybrid; GBP 50,000 - 75,000; Full time

Job summary

A leading company in AI is seeking a Senior AI Engineer in Edinburgh to redefine how legal teams operate. You'll develop AI-native experiences leveraging Large Language Models, collaborating with cross-functional teams to deliver impactful solutions. This hybrid role encourages ownership of mission-critical systems in a high-trust environment, ideal for those passionate about applying AI to real-world challenges.

Benefits

Competitive salary and meaningful equity

Ownership over mission-critical AI systems

Challenging product problems in law and AI

Collaborative environment with diverse expertise

Qualifications

• 4+ years in backend or AI-focused engineering.
• Experience with LLMs and frameworks like LangChain.
• Proven ability to design evaluation frameworks.

Responsibilities

• Build and maintain backend systems powering LLM features.
• Design evaluation pipelines to optimize model performance.
• Collaborate with legal teams to implement secure AI systems.

Skills

Python

Cloud Infrastructure

Evaluation Frameworks

Data Privacy

Scalable Architecture

Tools

LangChain

PyTorch

Scikit-learn

Job description

Role: AI Engineer
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

• Interest in prompt engineering, user intent modeling, or AI agent design

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.