Lead Machine Learning Engineer

As a Lead Machine Learning Engineer, you’ll join our growing team of world-class machine learning engineers and clinical experts to deploy machine learning models that help automate burdensome administrative clinical practices.

Our ML team’s work is focused on designing, deploying, and monitoring production models which extract or predict relevant clinical findings from structured and unstructured data sources.

Last but not least: People who succeed here are empathetic teammates who are candid, kind, caring, and embody our core values and principles. We believe that diverse, inclusive teams make the most impactful work. Cohere is deeply invested in ensuring that we have a supportive, growth-oriented environment that works for everyone.

What you’ll do:

• 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.

What you’ll need:

• 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

Bonus points for:

• 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

Pay & Perks:

💻 Fully remote opportunity with about 20% travel

🩺 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

The salary range for this position is $170,000.00 to $190,000.00 annually; as part of a total benefits package which includes health insurance, 401k and bonus. In accordance with state applicable laws, Cohere is required to provide a reasonable estimate of the compensation range for this role. Individual pay decisions are ultimately based on a number of factors, including but not limited to qualifications for the role, experience level, skillset, and internal alignment.