Note: The job is a remote job and is open to candidates in USA. UST is a technology company that delivers innovative, human-centered solutions for clients worldwide. The Machine Learning Engineer I will build and manage Databricks feature engineering pipelines, validation frameworks, feature stores, and Snowflake rule deployments. The role also involves monitoring model and rule performance and implementing CI/CD for data pipelines and model deployment.
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.
- Full-time, regular employees receive 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.
- Full-time, regular employees receive Company-paid basic life insurance, accidental death and disability insurance, and short- and long-term disability benefits for employees only.
- 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