Machine Learning Engineer

London, UK



£45,000 - £65,000


Apply now

Robin AI is an exciting, venture-backed technology startup whose mission is to dramatically reduce the cost and complexity of legal advice. Backed by top London based VCs, Forward Partners and Episode 1, Robin uses a combination of human and artificial intelligence to review and edit contracts and is building software to fundamentally change the legal industry forever.

Robin was established in March 2019 by CEO Richard Robinson (a former City lawyer) and CTO James Clough (a theoretical physicist who works in Machine Learning). They have since assembled a talented and diverse team of legal, software and machine learning engineers to build the next generation of contracting tools.

But our ambitions go way beyond reviewing contracts – we want to build technology that changes the way companies all over the world do deals. The legal industry has, thus far, been slow to embrace artificial intelligence. In part this is because the law is a naturally conservative industry, one we think is ready for disruption. But it is also because existing technology has struggled to tackle some of the unique challenges this field presents. Data is usually confidential and so datasets are limited in size. And the legal and financial stakes are high, meaning that high model accuracy and interpretability are especially important. Traditional NLP approaches can under-perform on legal tasks due to the unusual and precise meanings of particular words and phrases. All of these factors mean that we at Robin AI face a unique set of technical challenges that can only be tackled by creative and cross-disciplinary thinking.

The Role

As a senior machine learning engineer at Robin, you will be a leader in our ongoing effort to overcome these challenges. You will design and implement the taxonomy and models we use to understand the meaning of legal documents, and build scalable training pipelines. By keeping up to date with the latest research you will bring state of the art methods into our customer-facing product as we expand our service across different industries and contract types.As well as developing the technology that enables us to comprehend the unstructured documents currently used in legal services you will also help us create products to draft, edit and query more structured documents, automatically parsing and editing their legal meaning. Machine learning plays a pivotal role in Robin’s core business because it is the best way to take the millions of contracts we want to analyse and make them structured enough to be manipulated by AI.

Tech Stack

Our machine learning pipeline is written in PyTorch, using HuggingFace Transformers tokenisers and models as a base. We also use spaCy for natural language processing and have built our own libraries for integrating with technologies used by legal teams. Models are trained and deployed on cloud services (GCP and AWS). The back-end services of our product are implemented with Python web frameworks (Django, Flask) and interact with our ML services using APIs written by the ML team.


  • Own and build our machine learning infrastructure - for both offline training and online inference. Ensure that this infrastructure can scale to service our rapidly growing business.
  • Keep up to take with the latest tools, methods and research, and bring cutting edge technologies into our products.
  • Delve into our datasets, and understand the complex environment our products operate in, so that you can improve the way we use ML to automate challenging work.
  • Mentor junior team members and become a leader in a growing ML team.

Your Skills

  • You have experience designing, implementing and training machine learning models using a modern Python ML framework.
  • You write high quality, well-tested code, and have experience putting machine learning models into real products.
  • You are a strong communicator who can both explain complex technical ideas to others, and can productively engage with experts in other domains to understand the technical needs of the company as a whole.


  • You have a background working in a start-up or similar size organisation
  • You are familiar with Python frameworks (eg. Django, Flask)
  • You have experience applying ML to natural language processing tasks


£45,000 - £65,000
Competitive salary
Meaningful equity in the business
Flexible working, mix working from home & in our London office
Focus on continual learning and development
Desk customisation budget and personal desk plant

We are deliberate about the culture we are building, and are committed to creating an inclusive environment that attracts the best talent from diverse backgrounds. Whether it’s helping curate the Robin Spotify playlist, discussing gender equality in tech, or taking a meditation break with our holistic health plan, we want you to find your space within the Robin community.

From our offices in Devonshire Square, you’ll have access to all of the WeWork amenities including co-working space, bike lockers, pool tables, table tennis, free barista-made coffee and free beer in the evenings.

DEI @ Robin AI

We want to break diversity, inclusion and equality barriers within the tech industry. In building our team, we welcome the unique contributions that you can bring, and welcome applications regardless of ethnicity, religion, sexual orientation, gender identity and expression, disability, education, opinions, nation of origin, languages spoken, age, pregnancy and maternity, or veteran’s status.

Apply now

Direct application

Upload failed. Max size for files is 10 MB.
(max file size 10MB)
Thank you! Your application has been received!
Oops! Something went wrong while submitting the form.