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

Requirements

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 anonymized data, and using them to train and fine-tune ML models
Experience with machine learning frameworks like TensorFlow, 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

Benefits

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 & ClassPass 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

Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools

machine learningML modelsbackend API servicesPythonSQLTensorFlowPyTorchScikit-Learndata sets creationgenerative AI

Soft Skills

collaborationcommunicationproblem-solvingadaptabilitycreativity