Senior Machine Learning Engineer

Yalent is partnering with a leading AI start-up in the legal field to find a highly skilled Senior Machine Learning Engineer to join their engineering team. In this role, you’ll directly support our client’s mission by developing and deploying innovative machine learning models, tackling complex data challenges, and enhancing their AI capabilities. As a Senior ML Engineer, you’ll be essential to scaling our client’s technology and driving impactful solutions.

• Collaborate with stakeholders to identify and map business problems to ML solutions
• Design, develop, and implement ML models with a focus on NLP and generative AI applications
• Curate, clean, and prepare data for model development and training
• Create and maintain ETL jobs for data processing
• Conduct rapid prototyping of ML solutions to quickly iterate on ideas
• Stay current with the latest advancements in ML, particularly in NLP and generative AI
• Collaborate with the platform engineering team to integrate ML solutions into the overall product architecture
• Implement data flywheels to continuously improve ML features through increased usage
• Define and implement ML performance metrics
• Contribute to the product roadmap with ML-driven feature ideas
• Be an essential technical contributor at a Series B-stage company as it scales

• 4-5 years of experience in applied machine learning or related field
• Strong proficiency in Python and its ML/data science libraries
• Extensive experience with NLP techniques and generative AI technologies
• Experience with LLMs and both text-to-text and text-to-image generative models
• Proficiency in working with large datasets and creating ETL processes
• Experience with version control systems (e.g., Git) and CI/CD practices
• Ability to work in a fast-paced, changing startup environment

• Competitive compensation packages with meaningful ownership
• Unlimited PTO
• 401k
• Wellness benefits, including a bundle of free therapy sessions
• Technology & Work from Home reimbursement
• Flexible work schedules