Staff AI Software Engineer – Hybrid ML in Production

iManage; City of London; Hybrid; GBP 80,000 - 100,000; Full time

Job summary

A leading technology firm in the United Kingdom is seeking an experienced AI Engineer. You will work at the intersection of cutting-edge machine learning and production systems, ensuring that legal documents are made accessible through AI. This role involves designing robust systems, collaborating directly with customers, and implementing feedback to improve solutions. Strong JVM expertise and problem-solving skills are required. The company supports flexible work arrangements and promotes a diverse workforce.

Benefits

Market-leading salary

Annual performance-based bonus

Flexible time off policy

BUPA private medical insurance

Enhanced parental leave

Qualifications

• 7+ years of engineering experience building and shipping production software.
• Deep expertise in JVM ecosystem with ability to build high-performance systems.
• Good judgment on applying AI vs traditional engineering.

Responsibilities

• Designing retrieval systems for legal documents accessible to AI.
• Working with customers for feedback and rapid iteration.
• Building production systems for high query loads.
• Developing frameworks to improve retrieval quality.

Skills

Problem-solving mindset

Self-directed

Team collaboration

Education

Bachelor's degree in Computer Science

Tools

JVM ecosystem (Java/Scala)

Kubernetes

Azure cloud services

Job description

A leading technology firm in the United Kingdom is seeking an experienced AI Engineer. You will work at the intersection of cutting-edge machine learning and production systems, ensuring that legal documents are made accessible through AI. This role involves designing robust systems, collaborating directly with customers, and implementing feedback to improve solutions. Strong JVM expertise and problem-solving skills are required. The company supports flexible work arrangements and promotes a diverse workforce.