AI/ML Engineer

Optum is a global organization that delivers care, aided by technology to help millions of people live healthier lives. The work you do with our team will directly improve health outcomes by connecting people with the care, pharmacy benefits, data and resources they need to feel their best. Here, you will find a culture guided by inclusion, talented peers, comprehensive benefits and career development opportunities. Come make an impact on the communities we serve as you help us advance health optimization on a global scale. Join us to start Caring. Connecting. Growing together.  

Primary Responsibilities:

• Design & Build AI Solutions
  • Develop AI-driven solutions tailored to healthcare business challenges
  • Compare and evaluate multiple models and architectures to select the best fit for each use case
  • Quantize and optimize GenAI/LLM models for performance, cost, and accuracy
  • Implement Agentic AI frameworks to build autonomous, multi-step workflows for healthcare applications
  • Lead large, complex proof-of-concepts spanning on-premises and cloud environments for healthcare use cases
• End-to-End Product Development
  • Own the full lifecycle: data ingestion & transformation, feature engineering, analysis & modeling, deployment, testing, performance monitoring, observability, and documentation
  • Build robust RAG pipelines with vector databases for high-recall retrieval and LLM-driven generation
  • Establish, document, and evangelize best practices for machine learning, deep learning, and GenAI development
• LLM Excellence
  • Design, implement, and optimize prompts (including In-Context Learning) to improve reliability and task fidelity
  • Integrate Langfuse and LangSmith for LLM observability, tracing, evaluation, and continuous improvement of model behavior in production
• Technical Leadership
  • Prioritize and drive technology workstreams for ML, NLP, and GenAI products
  • Mentor and train scientists/engineers on complex AI/ML topics and code quality standards
  • Collaborate with cross-functional stakeholders to translate healthcare business needs into measurable AI outcomes
• Operational Excellence
  • Implement monitoring, testing, and maintenance practices to keep models robust, fair, and compliant

  • Manage the lifecycle and governance for a suite of NLP/ML models supporting multiple products

• Comply with the terms and conditions of the employment contract, company policies and procedures, and any and all directives (such as, but not limited to, transfer and/or re-assignment to different work locations, change in teams and/or work shifts, policies in regards to flexibility of work benefits and/or work environment, alternative work arrangements, and other decisions that may arise due to the changing business environment). The Company may adopt, vary or rescind these policies and directives in its absolute discretion and without any limitation (implied or otherwise) on its ability to do so

Required Qualifications:

• Undergraduate degree or equivalent experience
• 4+ years applying computational algorithms and statistical methods to structured and unstructured data
• 3+ years with AI Cloud services (e.g., Azure ML, Databricks, Azure OpenAI), or equivalent on GCP/AWS
• 1+ years hands-on with LLMs, RAG (Retrieval-Augmented Generation), and Agentic AI frameworks
• Significant experience deploying scalable solutions end-to-end-from problem definition to successful product launch
• Practical experience in model performance tuning (latency, throughput, accuracy) including quantization and optimization
• Mandatory: Hands-on experience with Agentic AI frameworks such as Microsoft AutoGen, LangChain, LangGraph, and Semantic Kernel
• Solid Python development; expertise in machine learning and deep learning with TensorFlow or PyTorch (including distributed training)
• Deep subject-matter expertise in text processing, NLU, and SLU
• Proven ability to design effective prompts and optimize them for task performance; experience applying ICL
• Data engineering fundamentals: ingestion, transformation, and management
• Proven excellent communication, writing, and presentation skills
•  Proven ability to decompose complex problems into clear, implementable solutions and lead cross-functional delivery

Tools & Technologies

• Programming & DL: Python, TensorFlow, PyTorch, Hugging Face Transformers
• LLM Observability & Evaluation: Langfuse, LangSmith
• LLM & Orchestration: Azure OpenAI, LangChain, LangGraph, Semantic Kernel, Microsoft AutoGen
• RAG & Vector DBs: Azure AI Search, FAISS, Pinecone, Weaviate
• Cloud & Data: Azure ML, Databricks, Azure AI Foundry, Azure Data Factory (plus equivalents on GCP/AWS)
• Infra & Ops: Docker, Kubernetes, CI/CD, Terraform (preferred)
• Agentic AI frameworks: ADK, Autogen, LangChain, LangGraph

At UnitedHealth Group, our mission is to help people live healthier lives and make the health system work better for everyone. We believe everyone-of every race, gender, sexuality, age, location and income-deserves the opportunity to live their healthiest life. Today, however, there are still far too many barriers to good health which are disproportionately experienced by people of color, historically marginalized groups and those with lower incomes. We are committed to mitigating our impact on the environment and enabling and delivering equitable care that addresses health disparities and improves health outcomes - an enterprise priority reflected in our mission.