Senior AI/ML Software Engineer

2 days ago


San Francisco, California, United States Rocket Lawyer Full time $150,000 - $256,000
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 Senior 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

  • Rapidly prototype product features to explore new AI capabilities, while writing clean, understandable code
  • Deliver generative AI features to production
  • Mentor junior AI developers
  • 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
  • Build, train, fine-tune, deploy and maintain ML models for AI features
  • Keep up-to-date with industry trends around generative AI and machine learning

What you'll need

  • 7+ years of experience writing ML code for launched products
  • Experience leading a small team of developers
  • Experience writing production features leveraging LLMs and generative AI
  • 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
  • Strong programming skills in Python
Highly Desired
  • Experience with Pydantic AI
  • Experience with vector databases
  • Strong programming skills in NodeJS, Java, Kotlin, or similar languages
  • Experience or familiarity with AI model evaluation techniques
  • Interest in building user-facing products
  • Understanding of microservices architectures and distributed systems
  • Experience with building data pipelines using technologies such as Snowflake or Hive
  • Experience with machine learning frameworks like TensorFlow, PyTorch and Scikit-Learn
  • Strong programming skills in SQL

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 .

Compensation
Base salary range by location:
  • San Francisco Bay Area, CA: $189,000 - $256,000
  • California (outside of San Francisco Bay Area) and Colorado: $160,000 - $218,000
  • Utah, Arizona, and North Carolina: $150,000 - $205,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. $150,000—$256,000 USD

By applying for this position, your data will be processed as per Rocket Lawyer Privacy Policy. 



  • San Jose, California, United States Vectra AI Full time $150,000 - $250,000 per year

    About UsWe're an early-stage startup on a mission to reinvent cybersecurity with AI-native infrastructure. Our vision is to help defenders move faster than adversaries by combining large-scale data, modern AI, and intelligent automation. We're backed by top-tier investors and working with design partners across critical infrastructure and enterprise...


  • San Francisco, California, United States AI Fund Full time $150,000 - $250,000 per year

    About BasetenBaseten powers inference for the world's most dynamic AI companies, likeTHE ROLEAs a Senior Infrastructure Software Engineer at Baseten, you'll architect and lead development of our ML inference platform that powers production AI applications. You'll make key technical decisions for the infrastructure enabling developers to deploy, scale, and...

  • AI / ML Engineer

    5 days ago


    San Francisco, California, United States Neuron Factory Full time $200,000 - $250,000 per year

    About the CompanyWe're a software company on a mission to build the AI Co-Worker, a powerful platform designed to learn complex jobs and deploy intelligent agents to solve them at Scale.Our technology sits at the intersection of AI and human productivity, enabling businesses to automate sophisticated workflows like never before.About the RoleAs we scale, we...


  • San Francisco, California, United States Largeton Group Full time $180,000 - $250,000 per year

    Role OverviewLead the design, development, and implementation of the company's AI/ML platforms, infrastructure, and services.Drive technical vision, architecture, and strategy for AI/ML engineering needs, ensuring platform reliability, performance, and scalability.Provide technical leadership, guidance, mentorship, and foster a culture of continuous...


  • San Francisco, California, United States Multiscale AI Full time $120,000 - $180,000 per year

    About Multiscale AI:Multiscale AI provides advanced AI solutions for Semiconductor Manufacturing Process Optimization (SMPO). With offices in Atlanta, San Francisco, and Hyderabad, we help semiconductor manufacturers optimize fabs and achieve measurable results.The Role:We are seeking a Senior Software Engineer to join our Forward Deployed Engineering team....


  • San Francisco, California, United States FlowState AI Full time $150,000 - $200,000 per year

    Company DescriptionFlowstate AI is building next-generation AI agents that watch and analyze videos so humans don't have to. Our platform enables archival search, structured data extraction, content moderation, and generative applications, initially targeting enterprise clients in the media and entertainment industry. As a nimble, early-stage team, we have...


  • San Francisco, California, United States Sigma Computing Full time $240,000 - $270,000

    About the RoleAt Sigma, we're not just adding AI—we're building the future of how people work with data. Our platform already lets users explore billions of rows of data in seconds with a spreadsheet-like interface, analyze and present their data in workbooks, and build data apps and workflows. Now we're pushing further, applying AI to reshape how people...


  • San Francisco, California, United States Chime Full time $176,490 - $245,100 per year

    About The RoleChime's AI/ML team is building models, services, and platforms that transform how millions of users manage and grow their financial lives. We're looking for a Senior AI/ML Engineer with deep expertise in production-grade deep learning systems—someone who thrives at the intersection of cutting-edge research and real-world impact.As a Senior...


  • San Francisco, California, United States Glue Full time $125,000 - $175,000 per year

    About Glue Glue is not just another team chat app. We're building the first platform for agentic team chat—a workspace where AI agents and humans collaborate as peers. Our platform supports leading AI models (GPT-5, Claude, Gemini, and open-source alternatives) and integrates with thousands of apps through the Model Context Protocol (MCP), enabling...


  • San Francisco, California, United States Attentive Full time $170,000 - $230,000 per year

    Attentive is the AI-powered mobile marketing platform transforming the way brands personalize consumer engagement. Attentive enables marketers to craft tailored journeys for every subscriber, driving higher recurring revenue and maximizing campaign performance. Activating real-time data from multiple channels and advanced AI, the platform personalizes...