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 seeking a skilled and experienced Machine Learning Engineer to join our team.The ideal candidate will have a strong background in Python and PyTorch, along with 3-5 years of experience deploying ML/AI models to production. This role requires excellent analytics skills and a good working knowledge of Databricks. You will work closely with data scientists, clinicians, software engineers, and product teams to design, build, and optimize scalable machine learning solutions.

Responsibilities

·       Develop, train, and optimize machine learning models using PyTorch and other ML frameworks.

·       Deploy and maintain ML models in production environments, ensuring scalability, performance, and reliability.

·       Utilize Databricks for data processing, model training, model deployment, and pipeline optimization.

·       Deploy Retrieval-Augmented Generation (RAG) pipelines to production for improved AI-driven applications.

·       Collaborate with data engineers to design and implement ETL workflows and data pipelines.

·       Perform rigorous testing, validation, and monitoring of deployed models.

·       Optimize model inference for low latency and high throughput applications.

·       Work with stakeholders to translate business problems into ML solutions.

·       Stay up to date with the latest advancements in machine learning, deep learning, and AI deployment strategies.

Required Skills & Experience

Required Qualifications

·       Proficiency in Python and ML frameworks such as PyTorch.

·       3-5 years of experience deploying machine learning models to production.

·       Knowledge of MLflow for experiment tracking and model management.

·       Strong experience with Databricks for ML development and deployment.

·       Hands-on experience with MLOps, CI/CD pipelines, and cloud-based deployment (AWS, Azure, or GCP).

·       Solid understanding of data structures, algorithms, and software engineering principles.

·       Experience working with large-scale datasets and distributed computing frameworks.

·       Experience with deploying Retrieval-Augmented Generation (RAG) pipelines to production.

·       Excellent analytical and problem-solving skills.

·       Strong communication skills and ability to work in a collaborative team environment.

Preferred Qualifications

·       Experience deploying models in the healthcare domain.

·       Experience with feature engineering, data preprocessing, and model explainability.

·       Knowledge of containerization (Docker, Kubernetes) and workflow orchestration tools.

·       Familiarity with LLMs, NLP, or reinforcement learning is a plus.

Welldoc operates in ISO 13485 and MDSAP regulated and HITRUST and SOC 2 Type II compliant environments.  Therefore, it is expected that all employees will have either, prior experience working in  those environments or, will be trained to understand the requirements needed to work and support those requirements and culture as they relate to individual roles and responsibilities.

Required Education                                   

·       Bachelor’s degree in Computer Science, Data Science, Business Analytics, or a related field (Master’s preferred with 3-5 years of experience).

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 position is 100-140K/year with the expectation that most candidates will fall around the midpoint of the range. Candidates who exceed the required and preferred qualifications may be considered for an annual salary in excess of the budgeted 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 under 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.