Rocket Lawyer
Rocket Legal+ is the one membership for everything legal. Get legal and tax help – Anytime. Anywhere. #ConfidentlyLegal
Senior AI/ML Software Engineer, Remote
Location
California
Posted
121 days ago
Salary
$150K - $256K / year
Postgraduate Degree7 yrs expEnglishDistributed SystemsJavaKotlinMicroservicesNode.jsPythonPy TorchScikit LearnSQLTensorflow
Job Description
• 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
Job Requirements
- 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
- 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
- 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
