Lead AI/ML Engineer

Role Description

Join us as we work to create a thriving ecosystem that delivers accessible, high-quality, and sustainable healthcare for all. We are looking for a Lead Machine Learning Engineer who will lead the effort to apply Natural Language Understanding and Processing (NLUP) in the Healthcare Technology space.

• Lead the execution of research and enablement of NLUP initiatives, e.g. Large Language Model, Transformers, Retrieval Augmented Generation, GPT, Named Entity Recognition and Vectoring.
• Contribute to the overall NLUP strategy in applying such technology to our current pipelines and capability, e.g. Personalization, Recommendation, Search, Content Generation, Knowledge Graph, Summarization, Tagging, Topic and Intent Modeling.
• Lead the effort to collect data for training NLUP models, including working with developers for the Human-in-the-loop curator tool.
• Lead the effort to continuously train and publish NLUP inference models for downstream pipelines and services.
• Follow through on commitments and own the outcome.
• Mentor and coach team members.
• Understand and follow coding conventions, architecture patterns, and best practices.
• Participate in code reviews and perform peer code reviews to ensure quality standards.
• Proactively identify, propose, and justify new tech that will benefit the software product.
• Work collaboratively on a cross-functional team with a wide range of experience levels, e.g. ML Ops, Search, Engagement, Medical Informatics, Content Publishing, Platform, Legal & Security.
• Occasionally lead Agile planning and Scrum ceremonies.

Qualifications

• Bachelor's Degree in a relevant field such as Computer Science, Data Science, Operations Research, Math, Statistics, or equivalent.
• Ideally 5 years of professional experience in some or all aspects of designing, building, training, and evaluating Machine Learning Language Model.
• Hands-on data science/AI/ML engineering projects with proven results and outcomes.
• Demonstrated proficiency with Python (must) and Python ecosystem in data analysis and modeling tools.
• Strong experience in PyTorch, TensorFlow, Transformer (BERT and/or GPT) Fine Tuning, Vector Indexing Search, Large Language Model (LLM) with Retrieval Augmented Generation (RAG).
• Demonstrated experience with relational databases (e.g., SQL, Postgres).
• Passionate about learning and disseminating new knowledge.
• Experience with public cloud technologies, Amazon Web Services (AWS) is highly preferred.
• Familiarity with Linux/Unix/Shell environments.
• Exposure to Knowledge Graph (KG) in LLM is a plus.
• Apache SPARK experience is a plus.
• Apache Kafka experience is a plus.
• Effectively communicating complex technical information to non-technical audiences is a plus.

Requirements

• $148,000 - $252,000
• The base salary range shown reflects the full range for this role from minimum to maximum.
• Base pay depends on multiple factors, including job-related experience, relevant knowledge and skills, how your qualifications compare to others in similar roles, and geographical market rates.
• Base pay is only one part of our competitive Total Rewards package - depending on role eligibility, we offer both short and long-term incentives by way of an annual discretionary bonus plan, variable compensation plan, and equity plans.

Benefits

• Health and financial benefits.
• Perks specific to each location, including commuter support, employee assistance programs, tuition assistance, employee resource groups, and collaborative workspaces.
• Encouragement of a better work-life balance with flexibility in work environments.
• Sponsorship of events throughout the year, including book clubs, external speakers, and hackathons.
• A company culture based on learning, support of an engaged team, and an inclusive environment where all employees are valued.