AI/ML Software Engineer

Job Description

About Rocket Lawyer We believe everyone deserves access to affordable and simple legal services. Founded in 2008, Rocket Lawyer is the largest and most widely used online legal service platform in the world. With offices in North America and Europe, Rocket Lawyer has helped over 25 million people create over 50 million legal documents, and answer over 30,000 legal questions. We are in a unique position to enhance and expand the Rocket Lawyer platform to a scale, never seen before in the company’s history, to capture audiences worldwide. We are expanding our team to take on this challenge! About the Role Rocket Lawyer is seeking an experienced, passionate AI/ML software engineer for Rocket Copilot, our new AI assistant launched in December 2023. In this role, you will help our team deeply embed AI features across our platform. You will work in an agile, cross-functional team, working with multiple product teams as we build out AI features across Rocket Lawyer. You will design, implement, test, release and maintain new AI features, working closely with product managers, UX designers, data engineers, QA engineers, and others. We value a fun, collaborative, team-oriented work environment, where we celebrate our accomplishments.

Responsibilities

• Build, train, fine-tune, deploy and maintain ML models for AI features
• Create, monitor and troubleshoot backend API services to allow Rocket Lawyer product frontends to interact with ML models
• Work closely with multiple product teams and cross-functional teammates to quickly integrate AI into existing and new Rocket Lawyer products
• Keep up-to-date with industry trends around generative AI and machine learning
• Rapidly prototype product features to explore new AI capabilities, while writing clean, understandable code

Experience and Background

• 2+ years of experience writing ML code for launched products
• Understanding of the state of the art in generative AI, including RAG, LoRA, RLHF and others
• Experience creating data sets from proprietary anonymized data, and using them to train and fine-tune ML models
• Experience with machine learning frameworks like TensorFlow or PyTorch
• Strong programming skills in Python, Java, Kotlin, NodeJS or similar languages
• BS, MS, or PhD in Computer Science, Mathematics, Statistics, or other quantitative fields, or related work experience

Highly Desired

• Experience with vector databases
• Experience writing production features leveraging LLMs and generative AI
• Experience with building data pipelines using technologies such as Snowflake or Hive
• Experience or familiarity with AI model evaluation techniques
• Interest in building user-facing products
• Understanding of microservices architectures and distributed systems

£63,000 - £90,000 a year Apply for this job