[Remote] Machine Learning Engineer

Note: The job is a remote job and is open to candidates in USA. UST is a mission-driven technology company that creates transformative, human-centered solutions for clients across more than 30 countries. The company is seeking a Machine Learning Engineer I to build feature engineering pipelines, validation frameworks, and feature stores using Databricks and Snowflake. The role also focuses on rule and model monitoring, Snowflake deployment, and CI/CD for data and model pipelines.

Responsibilities

• Feature engineering pipelines in Databricks (provider profiles, peer benchmarks, temporal velocity)
• Rules validation framework (RuleValidator, schema checks)
• Databricks feature store management (provider_features, peer_benchmarks tables with Z-ordering)
• Rule performance monitoring (precision, recall, F1 per rule)
• Snowflake rules deployment (UDFs, stored procedures)
• Model monitoring (drift detection via Databricks or custom dashboards)
• CI/CD for Data pipeline and model deployment (automated testing + promotion from staging to production)
• Feature store integration with models (Databricks Feature Store model serving)

Skills

• UST is searching for a Machine Learning Engineer with 3-5 years Databricks + Snowflake production experience
• ·       Python 3+ years
• ·       Databricks 1+ years (PySpark, Delta Lake, Unity Catalog)
• ·       Snowflake 3+ years (SQL expert, UDFs, stored procedures), Feature Store (Databricks Feature Store with Z-ordering for fast lookups)
• ·       Healthcare claims data (CPT, ICD-10, modifiers) expert
• ·       YAML rules authoring or validation experience
• ·       MLflow experiment tracking (model versioning, A/B testing)
• ·       SQL (Databricks SQL + Snowflake), Git + GitHub
• ·       Must have built rule engines or feature pipelines
• ·       Databricks Certified Associate Data and ML Engineer (preferred)

Benefits

• Full-time, regular employees accrue a minimum of 10 days of paid vacation per year
• Full-time, regular employees receive 6 days of paid sick leave each year (pro-rated for new hires throughout the year)
• 10 paid holidays
• Full-time, regular employees are eligible for paid bereavement leave and jury duty
• Full-time, regular employees are eligible to participate in the Company’s 401(k) Retirement Plan with employer matching
• Full-time, regular employees and their dependents residing in the US are eligible for medical, dental, and vision insurance
• Company-paid basic life insurance for full-time, regular employees
• Company-paid accidental death and disability insurance for full-time, regular employees
• Company-paid short- and long-term disability benefits for full-time, regular employees
• Regular employees may purchase additional voluntary short-term disability benefits
• Regular employees may participate in a Health Savings Account (HSA)
• Regular employees may participate in a Flexible Spending Account (FSA) for healthcare, dependent child care, and/or commuting expenses as allowable under IRS guidelines
• Part-time employees receive 6 days of paid sick leave each year (pro-rated for new hires throughout the year)
• Part-time employees 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)
• Full-time temporary employees are eligible to participate in the Company’s 401(k) program with employer matching
• Full-time temporary employees 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

Company Overview

Company H1B Sponsorship