Staff ML Engineer

About the role

We are looking for a Staff Machine Learning Engineer to join our AI team and help us build cutting-edge legal solutions for some of the world's largest law firms. In this role, you will leverage your deep expertise in machine learning, natural language processing, and large language model (LLM) agents to push the boundaries of legal technology. You'll be collaborating closely with software engineers, and product managers to integrate advanced NLP techniques into our products.

What You'll Do:

• Innovative NLP Research: Research, evaluate, and implement state-of-the-art NLP methodologies and large language model approaches to drive product innovation and develop new functionalities.
• LLM Agents & Multi-Agent Systems: Design, develop, and deploy LLM agents and multi-agent systems to automate complex legal workflows and enhance user experiences.
• Advanced Technology Integration: Collaborate on projects that leverage emerging technologies - such as Retrieval-Augmented Generation (RAG) and Knowledge Graphs - to enhance our core product and explore new use cases.
• Cross-Functional Collaboration: Work closely with cross-functional teams to integrate advanced ML models and NLP solutions into our platform, ensuring they align with business objectives and provide tangible value.
• Strategic Innovation: Stay current with the latest trends and breakthroughs in NLP, machine learning, and multi-agent systems, and contribute ideas that shape the strategic direction of our AI initiatives.
• Mentor our ML team, promoting best practices in ML/NLP development, and elevate the technical bar across teams.

What you bring:

• Strategic Impact: Ability to translate complex technical opportunities into business outcomes and influence product strategy through data-driven insights and prototypes.
• Bias Toward Action: Self-directed, pragmatic, and motivated by real-world impact; capable of navigating ambiguity and driving initiatives from concept to production.
• Machine Learning & NLP Expertise: Strong understanding of machine learning and natural language processing with relevant commercial experience in building and deploying NLP solutions.
• Experience with LangGraph for Agentic Systems
• Cloud & Containerization Skills: Experience with the AWS cloud platform and containerization technologies (e.g., Docker, Kubernetes).
• Collaboration & Communication: Strong collaboration and communication skills to work effectively with cross-functional teams and articulate technical concepts to non-technical stakeholders.
• Proactivity & Continuous Improvement:Proactive in identifying problems, performance bottlenecks, and areas for improvement while taking pride in building and operating scalable, reliable, and secure systems.
• LLM Agents & Multi-Agent Systems: Proven experience in designing, deploying, and scaling large language model (LLM) agents and multi-agent systems to enhance NLP capabilities and automate complex workflows.

Most importantly, if you're someone with a can-do attitude who is willing to get their hands dirty to help us exceed our wildest expectations, then please do not hesitate to apply.

What We Offer:

• Competitive Compensation: A salary package aligned with your experience and impact.
• Meaningful Stock Options: Be rewarded for growing with the company
• Annual Bonus scheme: Eligible for the company bonus scheme.
• Real Impact & Growth: Be part of a scaling company where your work truly moves the needle.
• High-Performance Culture: Collaborate with ambitious, high-calibre teammates who raise the bar.
• Private Healthcare: Vitality healthcare, including Dental and Optical.
• Generous Time Off: 25 days of annual leave plus UK public holidays.
• Pension Plan: Competitive scheme to help plan for your future.
• Work From Anywhere Policy: Spend up to a month a year working abroad
• Enhanced Parental Leave: Inclusive policies that support working families.
• Top-Quality Equipment: Modern tech and ergonomic setups to help you do your best work.

About Definely

Definely is revolutionising how legal professionals access and understand information in complex documents. Our LegalTech solutions integrate directly into legal workflows, enabling teams to draft, review, and interpret contracts more efficiently, without breaking focus.

We're proud to be recognised among the Top 25 in Deloitte's UK Technology Fast 50 and backed by Revaia, Microsoft, Google, and Octopus Ventures. Trusted by top firms like A&O Shearman, Dentons, Deloitte, and Barclays, we're building tools that empower legal teams to work smarter and reduce risk.

At Definely, you'll be part of a mission-driven, collaborative, and ambitious team committed to innovation and growth.

Data Privacy Notice

By submitting your application, you agree that DEFEYENE LEGAL SOLUTIONS LIMITED ('Definely') may collect, process, and store your personal data as part of our recruitment process. We will use the information you provide to assess your qualifications for the role you are applying for and to communicate with you regarding your application.Your personal data will be stored for up to 12 months, after which it will be securely deleted unless we have another lawful basis to retain it. You have the right to access, correct, or request the deletion of your data at any time.For more details on how we handle your personal data and your rights, please send us an email to [email protected] and we will send your our privacy policy.

If you have any concerns about how your data is being processed, please do not hesitate to contact us.