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