Wordsmith AI Ltd

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.