Senior AI Engineer

Senior AI Engineer

Livingston Full-Time 43200 - 72000 £ / year (est.) No home office possible
W

At a Glance

  • Tasks: Build AI-driven features and maintain backend systems in a fast-paced startup.
  • Company: Join Scotland's fastest-growing startup, redefining legal team workflows with cutting-edge AI.
  • Benefits: Enjoy hybrid work, competitive salary, equity, and a collaborative environment.
  • Why this job: Be part of a mission-critical team tackling real-world legal challenges with innovative technology.
  • Qualifications: 4+ years in backend or AI engineering; strong Python skills and LLM experience required.
  • Other info: Opportunity to shape product and culture while working with top industry experts.

The predicted salary is between 43200 - 72000 £ per year.

JOIN SCOTLAND’s FASTEST GROWING STARTUP. 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
  • 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.

Senior AI Engineer employer: Wordsmith AI

Join a dynamic and innovative team at Scotland's fastest-growing startup, where you'll have the opportunity to work on cutting-edge AI solutions that are transforming the legal landscape. Our hybrid work model fosters a collaborative culture, allowing you to engage with talented professionals while enjoying the flexibility of remote work. With a strong focus on employee growth, competitive salaries, and meaningful equity, we empower our team members to take ownership of their projects and contribute to shaping both our product and company culture.
W

Contact Detail:

Wordsmith AI Recruiting Team

StudySmarter Expert Advice 🤫

We think this is how you could land Senior AI Engineer

✨Tip Number 1

Familiarise yourself with the latest advancements in Large Language Models (LLMs) and their applications in legal tech. This will not only help you understand the role better but also allow you to engage in meaningful conversations during interviews.

✨Tip Number 2

Network with professionals in the AI and legal tech sectors. Attend relevant meetups or webinars, and connect with current employees at StudySmarter on LinkedIn to gain insights into the company culture and expectations for the Senior AI Engineer role.

✨Tip Number 3

Prepare to discuss your hands-on experience with backend systems and APIs. Be ready to share specific examples of projects where you've implemented LLMs or designed evaluation frameworks, as this will demonstrate your practical knowledge and problem-solving skills.

✨Tip Number 4

Showcase your understanding of data privacy and system integrity in your discussions. Given the legal context of the role, being able to articulate how you would ensure compliance and security in AI systems will set you apart from other candidates.

We think you need these skills to ace Senior AI Engineer

Backend Development
Large Language Models (LLMs)
API Development
Python Programming
Evaluation Frameworks Design
Model Performance Tuning
Cloud Infrastructure Knowledge
Data Privacy Understanding
Scalable Architecture Design
Semantic Search Implementation
Document Analysis Techniques
Collaboration with Cross-Functional Teams
Automation of Model Evaluation
Experience with ML Frameworks (e.g. PyTorch, Scikit-learn)
Prompt Engineering

Some tips for your application 🫡

Tailor Your CV: Make sure your CV highlights relevant experience in AI engineering, particularly with Large Language Models and backend systems. Use specific examples that demonstrate your skills in programming, evaluation frameworks, and collaboration with cross-functional teams.

Craft a Compelling Cover Letter: In your cover letter, express your enthusiasm for the role and the company. Discuss how your background aligns with their mission to redefine legal workflows through AI. Mention any experience you have with legal tech or compliance tools, as this could set you apart.

Showcase Relevant Projects: If you have worked on projects involving LLMs, APIs, or semantic search, be sure to include these in your application. Describe your role in these projects and the impact they had, focusing on metrics or outcomes where possible.

Highlight Continuous Learning: Given the fast-paced nature of AI, mention any recent courses, certifications, or workshops you've completed related to AI, machine learning frameworks, or prompt engineering. This shows your commitment to staying updated in the field.

How to prepare for a job interview at Wordsmith AI

✨Showcase Your Technical Skills

Be prepared to discuss your experience with backend systems and LLMs in detail. Highlight specific projects where you've implemented evaluation frameworks or optimised model performance, as this will demonstrate your hands-on expertise.

✨Understand the Legal Context

Since the role involves working closely with legal teams, it’s beneficial to familiarise yourself with basic legal concepts and how AI can impact the legal field. This knowledge will help you engage more meaningfully during discussions.

✨Prepare for Problem-Solving Questions

Expect to tackle technical challenges or case studies during the interview. Practice articulating your thought process clearly, as this will showcase your problem-solving skills and ability to think on your feet.

✨Demonstrate Collaboration Skills

The role requires close collaboration with various teams. Be ready to share examples of how you've successfully worked with cross-functional teams in the past, particularly in high-impact projects that required input from multiple stakeholders.

W
Similar positions in other companies
UK’s top job board for Gen Z
discover-jobs-cta
Discover now
>