Note: The job is a remote job and is open to candidates in USA. UST is a mission-driven group that transforms lives through technology and is seeking an ML Platform Engineer. The role involves building MCP Servers for CIE and integrating various components to enhance machine learning model deployment and explainability.
Responsibilities
- Integrate with AgentCore Gateway (JWT/mTLS authentication, entitlement checks, routing)
- Deploy BHI custom models to Databricks Model Serving + Snowflake (UDFs, external functions)
- AWS Bedrock integration for LLM explainability via AI Marketplace subscription
- PHI guardrails implementation (pre-request scanning, post-response masking for healthcare data)
- MLflow model registry (versioning, A/B testing, rollback)
- Build CI/CD for model deployment, Feature store integration (Databricks Feature Store MCP tools)
- OpenTelemetry instrumentation (usage events emission to AI Marketplace metering)
- MCP tool manifest registration (score_claim, explain_detection, provider_profile tools)
Skills
- Python 5+ years
- Familiarity with MCP Protocol (Model Context Protocol) - build MCP servers that expose tools, MLOps/MLflow 3+ years (model registry, serving) on AWS and Databricks
- Databricks Model Serving production (REST API endpoints)
- Snowflake UDF + external function deployment
- OAuth2/JWT authentication + mTLS certificate handling, AWS Bedrock (LLM integration for explainability)
- AgentCore Gateway integration (JWT validation, entitlement enforcement, tool routing)
- PHI/HIPAA compliance (pre/post-request PHI scanning)
- OpenTelemetry (usage event emission, distributed tracing)
- REST API development (MCP tool endpoints)
- Docker/containerization
- Monitoring (Databricks, CloudWatch)
- Git + GitHub + CI/CD
- Must have 3-5 years MLOps with AWS + Databricks + Snowflake
- Must understand MCP Protocol or willingness to learn (AWS AgentCore documentation)
- Must have worked with OAuth2-secured APIs
- AWS/ Databricks Certified
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
- Eligible for paid bereavement leave and jury duty
- Eligible to participate in the Company’s 401(k) Retirement Plan with employer matching
- Eligible for medical, dental, and vision insurance
- 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
- Participate in a Health Savings Account (HSA)
- 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)
- 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