AI/ML Software Engineer (Remote in Arizona)

Rocket Lawyer
Jackson, AZ
Category Data Analyst
Remote
Job Description
Rocket Lawyer is seeking an experienced AI/ML Software Engineer to join their growing team working with AI features on the Rocket Copilot platform. The role involves designing, implementing, testing, and maintaining AI features, working in an agile environment with multiple product teams. The ideal candidate will have a strong background in generative AI, experience with LLMs and RAG, and a passion for building innovative AI solutions.

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
  • Strong programming skills in Python
  • High Desired: Experience with Pydantic AI, vector databases, and strong programming skills in NodeJS, Java, Kotlin, or similar languages
  • Experience or familiarity with AI model evaluation techniques
  • Strong programming skills in SQL
  • 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 NodeJS, Java, Kotlin, or similar languages
  • Experience or familiarity with AI model evaluation techniques

Benefits

  • Comprehensive health plans
  • 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
  • Financial Wellbeing & Student Loan Program access
  • Pre-tax Commuter/Transit Benefits
  • Free Rocket Lawyer account
]]>