AI/ML Engineer - AIML, Java Full Stack

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.

We are seeking an experienced Lead AI Engineer with over 7 years of expertise in designing, building, and scaling enterprise-grade AI systems. This role requires deep hands-on experience in Generative AI, Large Language Models (LLMs), LangChain/LangGraph, and cloud-native AI platforms such as AWS Bedrock, Azure AI Foundry.

In addition to strong GenAI capabilities (80%), the ideal candidate brings foundational experience in Full stack engineering (20%) to support robust, end-to-end AI solution development. This role involves technical leadership, architecture ownership, and mentoring engineers while collaborating cross-functionally to deliver secure, scalable, and responsible GenAI solutions.

Primary Responsibilities:

• Generative AI & LLM Engineering
  • Architect, design, and develop advanced GenAI applications using LLMs, LangChain, LangGraph, and related frameworks
  • Build multi-agent workflows, custom tools, orchestrators, and RAG pipelines for enterprise use cases
  • Fine-tune, evaluate, and optimize LLMs for domain-specific applications
  • Deploy production-grade GenAI systems on AWS Bedrock, Azure AI Foundry
  • Integrate LLMs with enterprise systems, APIs, and microservices.
    Evaluate emerging GenAI models and technologies to guide platform strategy
• LLM Security & Responsible AI
  • Implement guardrails to prevent prompt injection, reduce hallucinations, and ensure safe outputs
  • Apply best practices in LLM security, including moderation layers, access control, and auditability

  • Ensure alignment with Responsible AI principles: fairness, transparency, explainability, and privacy

• Cloud-Native AI Development
  • Deploy and maintain AI/GenAI solutions on AWS, Azure
  • Leverage AI platform services including Bedrock
  • Integrate models via REST APIs, SDKs, and orchestration tools
• Full Stack
  • Work with modern full stack technologies, including React for frontend and Java for backend development, to support AI-driven applications
  • Build and optimise data ingestion, transformation, and processing workflows within React and Java-based platforms for AI/ML pipelines
  • Ensure data availability, quality, and performance for Retrieval Augmented Generation (RAG) and model training pipelines in full stack environments powered by React and Java.
• 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:

• 7+ years of experience in AI/ML engineering, data science, or applied machine learning roles
• Hands-on experience with LLMs, LangChain, LangGraph, and GenAI architectures
• Experience deploying GenAI systems using AWS Bedrock, Azure AI Foundry
• Solid understanding of LLM security, prompt engineering, and Responsible AI principles
• Solid proficiency in Python, ML/NLP libraries, and AI development frameworks

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. #AIForBetterHealthcare