Reports to: CTO (permanent role) - No Visa Sponsorship
The company
An early-stage legal tech SaaS business. Its platform automates document-heavy legal workflows for law firms and institutional clients. It has product-market fit, a growing customer base and a lean engineering team.
The role
You will lead the development of AI features in the core product. This is a hands-on role.
Design and build ML models for document parsing, search, classification and summarisation
Integrate LLMs into product workflows with guardrails, evaluation and monitoring
Optimise model serving and inference for latency and cost on AWS (ECS, Lambda, S3)
Build data pipelines for training and evaluation (PostgreSQL, Redis, OpenSearch)
Work with engineering, product and QA in Scrum squads, with CI/CD and infrastructure as code already in place
Requirements
5+ years as an AI/ML engineer delivering NLP and ML models into production SaaS
Production LLM experience: RAG, fine-tuning, evaluation, monitoring
Strong Python, including building APIs and integrating models into applications
NLP frameworks such as Hugging Face Transformers, PyTorch, spaCy or TensorFlow
AWS and distributed systems
Data pipelines and orchestration for model training and evaluation
Nice to have
Legal tech or another document-heavy domain
Vector databases and semantic search (OpenSearch, Pinecone, FAISS)
Explainability and compliance work on deployed AI
Knows React or TypeScript
Right to work
No visa sponsorship is available. Applicants must already hold the full right to work in the UK.
We use essential cookies to make this site work. With your permission we would also like to use the optional cookies below. Nothing optional is set until you choose, and you can change your mind at any time from "Cookie settings" at the bottom of every page. Cookie policy.
Essential — keep the site working and secure, and remember the choice you make here. Always on.