AI / ML Engineer
unitedhealthgroup
Job Description
- Lead the design, development, and implementation of scalable AI/ML solutions, leveraging Python, SQL, pandas, numpy, and modern ML frameworks (TensorFlow, PyTorch, Scikit Learn)
- Oversee end to end ML workflows-from data preparation, feature engineering, EDA, and model development to evaluation, deployment, and continuous monitoring
- Guide model selection, optimization, and tuning, ensuring solutions balance performance, explainability, cost, and Responsible AI principles
- Build and maintain production grade ML pipelines with orchestration and experiment tracking tools such as MLflow, Feature Stores, ONNX, and vector databases
- Lead drift analysis, observability, and model health monitoring to ensure ongoing accuracy, reliability, and robustness in production environments
- Design and support scalable, multi cloud ML infrastructure (Azure/GCP primary, AWS optional), incorporating best practices for compute, storage, networking, and distributed data processing
- Collaborate with data engineering teams to integrate real time and batch pipelines built on Synapse/BigQuery, Dataflow/Databricks, and Kafka/EventHub/PubSub
- Work closely with engineering, product, data science, and UX to translate business requirements into actionable technical solutions and architecture patterns
-
Review and guide the creation of architecture diagrams, C4 models, ADRs, and technical specifications as part of engineering governance
- Comply with all applicable Company policies, procedures, and business directives, changes including those relating to work location, team assignments, work schedules, and flexible work arrangements