ML Data Engineer

Role Description

UST is searching for an ML Data Engineer with SQL (Databricks SQL + Snowflake), Git + GitHub skills. The opportunity includes:

• YAML rules schema design (medical_necessity, frequency_limits, coding_intensity)
• 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)
• CICD for Data pipeline and model deployment (automated sing + promotion from staging to production)
• Feature store integration with models (Databricks Feature Store model serving), Snowflake external function integration with AWS Lambda or Bedrock

What You Need

• Databricks Certified Associate Data and ML Engineer (preferred)
• 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

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.

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.
• Eligible to participate in the Company’s 401(k) Retirement Plan with employer matching.
• Eligible to participate in the following Company-paid Employee Only benefits: basic life insurance, accidental death and disability insurance, and short- and long-term disability benefits.
• 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) program with employer matching.
• 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.