Jobs

Staff Machine Learning Engineer


Job details
  • Robin AI
  • London
  • 1 week ago

About Robin AI

Robin AI is on amissionto 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 3 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 groundbreaking 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 that 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 optimize and scale model performance.
  • Develop internal tools to facilitate efficient dataset and evaluation results management, to streamline the 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 customized 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 productionizing LLMs (OAI, Anthropic, Mistral etc.) and/or SLMs (Llama, Phi etc.), using tools 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 prioritize promotions for high performers and help you to progress your career.

What’s it like working at Robin AI?

Our culture andvaluesattract 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.

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.

#J-18808-Ljbffr

Sign up for our newsletter

The latest news, articles, and resources, sent to your inbox weekly.

Similar Jobs

Senior / Staff Machine Learning Engineer

Job Description:  Arm's Machine Learning Group is seeking highly motivated and creative Software Engineers to join the Cambridge-based ML Content, Algorithms and Tools team! This Machine Learning Engineer role focuses on advancing the field of AI by optimizing and deploying pioneering models, particularly Large Language Models (LLMs) and Generative AI...

Cambridge

Senior / Staff Machine Learning Engineer - Cutting-edge AI innovations with LLM focus (Apply in minutes)

**Job Description:**Arms Machine Learning Group is seeking highly motivated and creative SoftwareEngineers to join the Cambridge-based ML Content, Algorithms and Tools team!This Machine Learning Engineer role focuses on advancing the field of AI byoptimizing and deploying pioneering models, particularly Large Language Models(LLMs) and Generative AI algorithms. This involves deep analysis...

ARM Cambridge

Senior / Staff Machine Learning Engineer (Apply in minutes)

Job Description:Arms Machine Learning Group is seeking highly motivated and creative SoftwareEngineers to join the Cambridge-based ML Content, Algorithms and Tools team!This Machine Learning Engineer role focuses on advancing the field of AI byoptimizing and deploying pioneering models, particularly Large Language Models(LLMs) and Generative AI algorithms. This involves deep analysis...

ARM Newton

Harnham | Staff Machine Learning Engineer

Staff Machine Learning EngineerLondon/Remote UK£125,000-£145,000This global gaming company are the leaders in their field and ahead of the game when it comes to AI & innovation. You'll be joining a small, international team of AI experts focusing on working with multi-modal data using Computer Vision & Generative AI techniquesTHE ROLEAs...

Harnham London

Senior / Staff Machine Learning Engineer

Job Description: Arm's Machine Learning Group is seeking highly motivated and creative Software Engineers to join the Cambridge-based ML Content, Algorithms and Tools team This Machine Learning Engineer role focuses on advancing the field of AI by optimizing and deploying pioneering models, particularly Large Language Models (LLMs) and Generative AI...

ARM Cambridge

Machine Learning Engineer

We have an exciting opportunity available for a Machine Learning Engineer (Data science / Deep learning) to join the Cancer Dynamics Laboratory. The post holder would work across multiple on-going projects of the lab, including a dedicated role within the MANIFEST project (“Multiomic ANalysis of Immunotherapy Features Evidencing Success and ...

St. Pancras and Somers Town