Staff Machine Learning Engineer

About Robin AI

Robin AI is on a mission to rebuild the legal industry — starting with making contracts simple for everyone. We are a pioneer in Legal AI, built on proprietary models, licensed data, and deep partnerships with Anthropic and AWS. Since 2019, we’ve expanded our footprint to 4 continents and have been supporting many of the world’s most successful businesses, including GE, Pfizer, KPMG, and UBS.

About the role

Robin AI is looking for Staff ML Engineers to join our team in developing ground breaking legal technology products. You will work at the exciting intersection of law and technology, collaborating with lawyers, AI researchers and product managers in order to help build industry-leading legal AI products.

We are looking for someone with a keen interest in legal AI, who wants to apply their strong ML Engineering background to build tools which can directly impact our customers.

What You’ll do

• Accelerate our AI & Research team by designing and implementing scalable infrastructure to support rapid experimentation.

• Collaborate with cross-functional teams to enhance experiment design, execution, and implementation of state-of-the-art AI models.

• Implement and adapt existing research methodologies to optimise and scale model performance.

• Develop internal tools to facilitate efficient dataset and evaluation results management, to streamline Legal Engineering team’s processes and evaluations.

• Lead the design and execution of online evaluation frameworks.

• Create and streamline end-to-end training, fine-tuning, distillation, and deployment pipelines to support customised solutions.

• Collaborate with Backend Engineers to integrate our research output into a cohesive system, enhancing feedback loops and overall functionality.

Preferred Qualifications

• Master’s degree in Computer Science, Machine Learning, or a related field.

• Extensive programming experience in Python and experience with machine learning frameworks such as TensorFlow, PyTorch, etc.

• Proven experience in machine learning engineering, with a focus on building robust, scalable systems, and strong competence in CI/CD for seamless integration and delivery of machine learning models.

• Experience with infrastructure and tooling development to support AI experimentation and deployment at scale.

• Strong collaboration skills and the ability to work effectively with cross-functional teams.

• Proven experience productionising LLMs (OAI, Anthropic, Mistral etc.) and/or SLMs (Llama, Phi etc.), using tool such as LangChain, etc.

• (preferred) Publication record in A* ML conferences and workshops.

• (preferred) Solid understanding of multi-agent systems, feedback loops, and message-passing infrastructures.

What’s in it for you

• Salary: £120,000 - £150,000

• Hybrid schedule: We offer a flexible working schedule. #LI-HYBRID

• Equity package: Generous equity scheme - everyone gets to be an owner of Robin AI!

• Annual leave: 25 days PTO, in addition to the bank holidays observed in the UK.

• Health and wellbeing: Comprehensive health insurance, mental healthcare, gym discounts and cycle to work scheme.

• Growth opportunities: We prioritise promotions for high performers and help you to progress your career.

This role is supported by AI Futures Grants, a UK Government scheme designed to help the next generation of AI leaders meet the costs of relocating to the UK. AI Futures Grants provide financial support to reimburse relocation costs such as visa fees, the immigration health surcharge and travel/subsistence expenses.

What’s it like working at Robin AI?

Our culture and values attract people who are creative, resourceful, and share our passion for excellence. At Robin, you're encouraged to push yourself and empowered to take risks. We support each other to think big, try new ideas, and navigate uncertainty. Whether you're at our headquarters or one of our worldwide offices, you'll find a world of opportunities to grow, thrive, and make a meaningful impact. See what life is like at Robin.

Diversity, Equity and Inclusion at Robin AI

We are committed to building one of the most diverse technology companies in the world. As of 2024, more than 30% of our employees come from ethnic minority backgrounds, and 51% of roles are held by women. We know that transforming the legal industry requires diverse perspectives, so we're creating an environment where innovation thrives through inclusion.