AI/ML Software Engineer (Remote in California)

Rocket Lawyer
San Francisco, CA
Category Data Analyst
Remote
Job Description
Rocket Lawyer is seeking an experienced AI/ML software engineer to join their team building AI features for their new AI assistant, Rocket Copilot. The role involves designing, implementing, testing, and maintaining AI features across the platform, working with cross-functional teams. They emphasize a collaborative and fun work environment with opportunities for rapid prototyping and mentoring.

Requirements

  • 5+ 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
  • Strong programming skills in Python
  • 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
  • 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 JavaScript

Benefits

  • Comprehensive health plans
  • Life insurance
  • Disability benefits
  • Supplemental Optional Life Insurance Benefits
  • FSA Options
  • HSA with Company Match
  • 401k program
  • Fertility Assistance and Planning options
  • Wellhub & ClassPass fitness platforms
  • Comprehensive Pet Insurance options
  • Financial Wellbeing & Student Loan Program access
  • Pre-tax Commuter/Transit Benefits
  • Free Rocket Lawyer account
  • Access to extensive legal documents
  • Competitive salary
  • Mental Health and Wellbeing resources
]]>