Rocket LawyerRemote
AI/ML Software Engineer
United States of America
Full Time
2 weeks ago
$165,000 - $190,000 USD
H1B Sponsor Likely
Key skills
Machine learningGenerative AIRetrieval-Augmented Generation (RAG)Low-Rank Adaptation (LoRA)Reinforcement Learning with Human Feedback (RLHF)Data set creation from anonymized dataTensorFlowPyTorchScikit-LearnPythonSQLAIMachine LearningMLRAGscikit-learn
About this role
Rocket Lawyer is the largest and most widely used online legal service platform in the world, seeking an experienced AI/ML software engineer for their new AI assistant, Rocket Copilot. The role involves embedding AI features across the platform, collaborating with cross-functional teams to design and implement new AI capabilities.
Responsibilities:
- Build, train, fine-tune, deploy and maintain ML models for AI features
- 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
- Keep up-to-date with industry trends around generative AI and machine learning
- Rapidly prototype product features to explore new AI capabilities, while writing clean, understandable code
Requirements:
- 5+ years of Experience writing ML code for launched products
- 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
- Experience with machine learning frameworks like TensorFlow, PyTorch and Scikit-Learn
- Strong programming skills in Python and SQL
- BS, MS, or PhD in Computer Science, Mathematics, Statistics, or other quantitative fields, or related work experience