Senior AI/ML Software Engineer (Remote in Colorado)
About The Position
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.
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
Nice To Haves
- 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
Responsibilities
- 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
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
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Mid Level
Education Level
No Education Listed
Number of Employees
251-500 employees
