Staff Machine Learning Engineer

Company & Culture

Welldoc is at the forefront of digital health, driven by a powerful mission: empowering better cardiometabolic health through AI-powered, personalized digital tech, with a vision to be the leading advanced AI digital technology partner across the healthcare industry. We're a team passionate about leveraging cutting-edge science to improve lives, united by core values of collaborative innovation, accountability to excellence, customer focus, efficiency, and unwavering integrity, quality, and safety.

At Welldoc, you'll thrive in a collaborative and innovative environment where your contributions directly impact our mission. Recognized as a Great Place to Work for the past four years and named to Modern Healthcare’s Best Places to Work 2025, as well as being an industry thought leader featured at SXSW and in the Wall Street Journal and Economist, we invite you to make a real difference in healthcare with us.

Job Purpose

We are looking for a Staff Machine Learning Engineer to act as the "Technical North Star" for our ML initiatives. In this role, you will architect, build, and scale robust, end-to-end machine learning pipelines—spanning classical ML to Generative AI—that support real-time clinical decision-making.

You will personally own the application architecture and technical standards for the ML team. You will serve as the primary technical bridge to our Bangalore engineering center, ensuring architectural consistency and engineering excellence across our global team.

Responsibilities

• Architect & Build: Design and implement robust, scalable Machine Learning pipelines (for both Classical ML and GenAI) that power our clinical decision-making. You will be hands-on with the code, defining the patterns for how we build, test, and deploy models.
• Platform Partnership: Collaborate with the Platform Engineering & Data team (India) to define the requirements for our underlying ML infrastructure. You will act as the "Lead Customer" of the platform, ensuring the infrastructure evolves to meet the specific needs of our complex US healthcare datasets and compliance requirements.
• Set Technical Standards: Act as the primary technical authority for the ML team. You will institute code quality standards, lead deep-dive code reviews, and drive the adoption of best coding practices across both US and Bangalore teams.
• MLOps & CI/CD Ownership: Define and implement the complete MLOps lifecycle. Architect CI/CD pipelines specifically for AI/ML projects to ensure automated, reliable, and efficient model training and deployment.
• Global Technical Alignment: Act as the primary technical bridge to the Bangalore ML team. You will foster a unified engineering culture by conducting regular technical syncs and traveling to the India office ( 4-5 times per year) to ensure process and architectural alignment.
• Production Deployment: Drive production deployment, monitoring, scaling, and maintenance of statistical, deep learning, and generative AI (LLM/RAG) models, ensuring high availability and performance.
• Governance & Compliance: Create and maintain data governance SOPs to ensure all models and processes conform to strict regulatory standards (e.g., ISO 13485, HIPAA, GDPR, SOC2).

Required Skills & Experience

• Experience: 7+ years of hands-on software engineering experience, with at least 4+ years dedicated to Machine Learning engineering in a production environment.
• The "Architect" Mindset: Proven track record of taking ML systems from "notebook prototype" to "highly scalable production service." You think about latency, inference costs, and system reliability.
• Technical Stack: Deep proficiency in Python and the modern ML stack (PyTorch, TensorFlow, Scikit-learn). Extensive experience with Databricks and Spark for data lake architecture.
• Generative AI: Proven experience deploying Generative AI models, including LLM and RAG applications, in a production environment.
• MLOps Mastery: Strong, hands-on understanding of MLOps principles: model versioning, drift monitoring, feature stores, and automated retraining.
• CI/CD: Experience designing and maintaining CI/CD pipelines using tools like Jenkins, Git, or Azure DevOps.
• Regulated Environment: Experience working in HealthTech, FinTech, or other highly regulated industries (ISO 13485, HITRUST, SOC 2) is a strong plus.
• Global Collaboration: Willingness and ability to travel internationally (to India) 4-5 times per year to collaborate with the distributed team.

Required Education                                   

• Master's or PhD degree in Computer Science, Engineering, Data Science, or a related technical field.

Compensation & Benefits

Welldoc offers a competitive compensation package which, in addition to salary, includes generous PTO, medical insurance, dental insurance, vision care, life and disability insurance, retirement benefits. and the opportunity to participate in health savings accounts and/or dependent care accounts. The anticipated salary range for this role is $175,000 - $225,000 per year, with the expectation that candidates meeting the minimum qualifications of the role be eligible for up to the midpoint of the range.

Upon receipt of a conditional offer of employment, you may be required to complete and clear a multi-panel drug screening process.  This screening is in connection with requirements set by certain of the Company’s customers, with which you may be working in this role, and will at all times be administered all applicable laws. 

Welldoc is an equal opportunity employer and prohibits discrimination and harassment of any kind. We offer an inclusive workplace and will not tolerate discrimination against any job candidate or employee due to age, race, religion, color, ethnicity, national origin, gender, gender identity/expression, sexual orientation, membership in an employee organization, medical condition, family history, genetic information, veteran status, marital status or parental status.