AI​/ML Software Engineer

About Your 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 with multiple product teams to design, implement, test, release and maintain new AI features, collaborating closely with product managers, UX designers, data engineers, QA engineers, and others.

• 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

• 5+ 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 anonymised data, and using them to train and fine‑tune ML models
• Experience with machine learning frameworks like Tensor Flow, PyTorch and Scikit‑Learn
• Strong programming skills in Python and SQL
• BS, MS, or PhD in Computer Science, Mathematics, Statistics, or other quantitative fields, or related work experience

• Comprehensive health plans (including Medical, Dental and Vision insurance for full‑time employees)
• Unlimited PTO
• Competitive salary packages
• Life insurance
• Disability benefits
• Supplemental Optional Life Insurance Benefits
• FSA Options Optional
• HSA with Company Match
• 401k program with Company Match
• Fertility Assistance and Planning options
• Wellhub & Class Pass fitness platforms
• Comprehensive Pet Insurance options
• Financial Wellbeing & Student Loan Program access
• Access to additional Mental Health & Wellbeing resources
• Pre‑tax Commuter/Transit Benefits
• Free Rocket Lawyer account with online access to an extensive legal documents library and brilliant licensed attorneys at discounted rates

• Recruiter Phone Screen
• Role Assessment(s)
• Hiring Manager Interview
• Panel Interviews
• Final Interview

Base salary range by location:

• SF Bay Area: $165,000 - $190,000
• California: $155,000 - $180,000
• Colorado: $140,000 - $165,000
• Arizona/Utah/North Carolina: $132,000 - $155,000

$132,000—$190,000 USD

Rocket Lawyer is proudly committed to recruiting and retaining a diverse and inclusive workforce. As an Equal Opportunity Employer, we never discriminate based on race, religion, color, national origin, gender (including pregnancy, childbirth, or related medical conditions), sexual orientation, gender identity, gender expression, age, military or veteran status, status as an individual with a disability, or other applicable legally protected characteristics.

We particularly welcome applications from veterans and military spouses.

All your information will be kept confidential according to EEO guidelines. You may request reasonable accommodations by sending an email to