Machine Learning Engineer

The Role

UST is searching for a Machine Learning Engineer to design, build, and deploy ML models that drive insights and automation across our healthcare payer business — including claims, clinical, provider, and Medicare Advantage data.

What You’ll Do

• Design, build, and validate machine learning models addressing healthcare payer use cases (e.g., claims risk scoring, clinical outcome prediction, provider network analytics, Medicare Advantage risk adjustment).
• Develop and maintain feature engineering pipelines on Databricks, including feature stores, transformations, and reusable feature sets for model training and inference.
• Train, tune, and evaluate models using standard ML frameworks, tracking experiments and managing the model lifecycle (versioning, registry, lineage).
• Deploy models to production on Databricks (batch and/or real-time serving), integrating with downstream applications and monitoring systems.
• Monitor deployed models for performance, data drift, and degradation, and implement retraining or remediation workflows as needed.
• Apply data security and privacy best practices when handling PHI/PII and other sensitive healthcare data, in line with HIPAA and enterprise compliance requirements.
• Partner with data engineers, AI/agentic systems developers, product owners, and compliance stakeholders to translate business problems into ML solutions.
• Write clean, well-tested, production-quality code and contribute to shared ML tooling, standards, and documentation.

This position description identifies the responsibilities and tasks typically associated with the performance of the position. Other relevant essential functions may be required.

What You Need

• 2–4 years of hands-on experience in machine learning engineering or applied data science, covering the full lifecycle from feature development through model deployment.
• Strong proficiency in Python and common ML libraries/frameworks (e.g., scikit-learn, PyTorch, TensorFlow, MLflow).
• Practical experience building, training, and deploying models on Databricks, including feature engineering, experiment tracking, and model serving/deployment on the platform.
• Working knowledge of AWS as the underlying cloud infrastructure for data and ML workloads.
• Solid understanding of data security fundamentals — encryption, access controls, and secure handling of sensitive data — with comfort operating in a regulated enterprise environment.
• Healthcare background is a strong plus — familiarity with claims, clinical, provider, or Medicare Advantage data is valued but not required for strong ML engineering candidates willing to learn the domain.
• Solid software engineering fundamentals: version control, testing, CI/CD, and code review practices.
• Strong analytical and communication skills, with the ability to explain model behavior and tradeoffs to both technical and non-technical stakeholders.

Nice to Have

• Databricks Certification (e.g., Databricks Certified Machine Learning Associate/Professional).
• AWS Certification (e.g., AWS Certified Machine Learning – Specialty, AWS Certified Solutions Architect).
• Prior experience in health insurance, health plans, or another regulated industry, with exposure to HIPAA, SOC 2, or similar compliance frameworks.
• Experience with MLOps tooling (MLflow, model registries, CI/CD for ML) and monitoring/observability for production models.

Location & Compensation

Compensation can differ depending on factors including but not limited to the specific office location, role, skill set, education, and level of experience. UST provides a reasonable range of compensation for roles that may be hired in various U.S. markets as set forth below.

Role Location: Remote-US

Compensation Range: $82,000-$123,000

Who We Are

Benefits

Full-time, regular employees accrue a minimum of 10 days of paid vacation per year, receive 6 days of paid sick leave each year (pro-rated for new hires throughout the year), 10 paid holidays, and are eligible for paid bereavement leave and jury duty. They are eligible to participate in the Company’s 401(k) Retirement Plan with employer matching. They and their dependents residing in the US are eligible for medical, dental, and vision insurance, as well as the following Company-paid Employee Only benefits: basic life insurance, accidental death and disability insurance, and short- and long-term disability benefits. Regular employees may purchase additional voluntary short-term disability benefits, and participate in a Health Savings Account (HSA) as well as a Flexible Spending Account (FSA) for healthcare, dependent child care, and/or commuting expenses as allowable under IRS guidelines. Benefits offerings vary in Puerto Rico.

Part-time employees receive 6 days of paid sick leave each year (pro-rated for new hires throughout the year) and are eligible to participate in the Company’s 401(k) Retirement Plan with employer matching. Full-time temporary employees receive 6 days of paid sick leave each year (pro-rated for new hires throughout the year) and are eligible to participate in the Company’s 401(k) program with employer matching. They and their dependents residing in the US are eligible for medical, dental, and vision insurance. Part-time temporary employees receive 6 days of paid sick leave each year (pro-rated for new hires throughout the year). All US employees who work in a state or locality with more generous paid sick leave benefits than specified here will receive the benefit of those sick leave laws.

Equal Employment Opportunity Statement

UST is an Equal Opportunity Employer. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, status as a protected veteran, or any other applicable characteristics protected by law. We will consider qualified applicants with arrest or conviction records in accordance with state and local laws and “fair chance” ordinances. UST reserves the right to periodically redefine your roles and responsibilities based on the requirements of the organization and/or your performance.

#UST #CB #LI-VK5 #LI-Remote

Skills

Machine Learning Engineering, Python, Feature Store, PyTorch, Databricks, Healthcare Systems, MLOps, Scikit-learn