Machine Learning Engineer

Welldoc Inc • Columbia, MD

80d • $100 - $140

About The Position

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

Requirements

• 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.
• Bachelor’s degree in Computer Science, Data Science, Business Analytics, or a related field (Master’s preferred with 3-5 years of experience).

Nice To Haves

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

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.

Benefits

• generous PTO
• medical insurance
• dental insurance
• vision care
• life and disability insurance
• retirement benefits
• opportunity to participate in health savings accounts and/or dependent care accounts

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Number of Employees: 101-250 employees