Easy Apply
Easy Apply
The AI/ML Software Engineer will design, implement, and maintain AI features and ML models for Rocket Lawyer's platform, working closely with cross-functional teams.
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, South America, and Europe, Rocket Lawyer has helped over 30 million people create over 50 million legal documents, and get their legal questions answered.
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 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, 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.
How you will make a difference day to day
- 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
What you’ll need
- 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 & Perks
- 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
Interview Process:
- Recruiter Phone Screen
- Role Assessment(s)
- Hiring Manager Interview
- Panel Interviews
- Final Interview
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 [email protected].
Compensation
Base salary range by location:
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
Actual compensation packages are determined by various factors unique to each candidate, including but not limited to skill set, depth of experience, certifications, specific work location, and performance during the interview process.
$132,000—$190,000 USD
By applying for this position, your data will be processed as per Rocket Lawyer Privacy Policy.
Top Skills
Python
PyTorch
Scikit-Learn
SQL
TensorFlow
Similar Jobs
Database • Analytics
Build AI/ML-powered features across ClickHouse Cloud, designing scalable APIs, integrating inference systems and vector DBs, implementing frontend interfaces, and ensuring production-grade monitoring, versioning, and observability for model deployments.
Top Skills:
Typescript,Javascript,React,Python,Clickhouse,Clickhouse Cloud,Aws,Azure,Gcp,Vector Databases,Inference Apis
Software
As a Software Engineer, AI/ML GenAI, you will design and implement AI features, manage embeddings, and optimize AI systems for production deployment while collaborating with product and design teams.
Top Skills:
AWSCeleryDockerFastapiGCPNode.jsPythonRuby On RailsSQLTypescript
Healthtech
The Senior Software Engineer I role involves developing scalable solutions, mentoring junior engineers, and collaborating on technical roadmaps within the healthcare industry.
Top Skills:
AWSAzureGCPMachine Learning AlgorithmsSQLStatistical Data TechniquesWeb Application Development
What you need to know about the Colorado Tech Scene
With a business-friendly climate and research universities like CU Boulder and Colorado State, Colorado has made a name for itself as a startup ecosystem. The state boasts a skilled workforce and high quality of life thanks to its affordable housing, vibrant cultural scene and unparalleled opportunities for outdoor recreation. Colorado is also home to the National Renewable Energy Laboratory, helping cement its status as a hub for renewable energy innovation.
Key Facts About Colorado Tech
- Number of Tech Workers: 260,000; 8.5% of overall workforce (2024 CompTIA survey)
- Major Tech Employers: Lockheed Martin, Century Link, Comcast, BAE Systems, Level 3
- Key Industries: Software, artificial intelligence, aerospace, e-commerce, fintech, healthtech
- Funding Landscape: $4.9 billion in VC funding in 2024 (Pitchbook)
- Notable Investors: Access Venture Partners, Ridgeline Ventures, Techstars, Blackhorn Ventures
- Research Centers and Universities: Colorado School of Mines, University of Colorado Boulder, University of Denver, Colorado State University, Mesa Laboratory, Space Science Institute, National Center for Atmospheric Research, National Renewable Energy Laboratory, Gottlieb Institute
