Lead Machine Learning Engineer

Summary

Join Cohere Health's growing team as a Lead Machine Learning Engineer to deploy machine learning models automating clinical practices. You will design, develop, and deploy advanced machine learning algorithms, build scalable production systems, and maintain reusable codebases. Collaboration with various stakeholders, including clinicians and product managers, is crucial. A master's degree and 7+ years of applied ML experience or a PhD and 5+ years of experience are required. The role offers a fully remote opportunity with benefits including comprehensive insurance, a 401k plan, generous PTO, and paid parental leave.

Requirements

• At least 7+ years of experience in applied ML with a masters degree or 5+ years with a PhD, ideally in a senior or lead role in a large technology or healthcare company
• Clear understanding of model building, model maintenance and the measures that optimize models for production use
• Understand experimental design and can independently perform collection, measurement, and interpretation of results
• Expert in Python with experience in deep learning frameworks (e.g., PyTorch)
• Hands on experience building deep learning models (e.g., transformers) for NLP tasks
• MS or PhD required

Responsibilities

• Design, develop and deploy advanced machine learning algorithms for retrieval, classification and generative use cases
• Develop reliable and scalable production machine learning systems
• Write and maintain reusable codebases to perform data preprocessing, model training, evaluation, and deployment
• Provide software and machine learning best practice guidance to junior engineers
• Work cross-functionally with diverse stakeholders, including product managers, clinicians, and technical leadership to align ML solutions with Cohere’s core objectives

Preferred Qualifications

• Experience with generative AI or Retrieval Augmented Generation (RAG)
• Experience working with unstructured healthcare data (e.g., clinical notes)
• Experience with OCR or image-based document understanding techniques
• Hands on experience using AWS tools such as SageMaker Studio

Benefits

• Medical, dental, vision, life, disability insurance, and Employee Assistance Program
• 401K retirement plan with company match; flexible spending and health savings account
• Up to 184 hours (23 days) of PTO per year + company holidays
• Up to 14 weeks of paid parental leave
• Pet insurance
• Fully remote opportunity with about 20% travel