Skip to main content
Posted 14 June, 2026

Lead AI/ML Engineer

UnitedHealth Group
Bengaluru, KA, IN Full Time
Reference: e419785982de2a55

Job Description Optum is a global organization that delivers care, aided by technology to help millions of people live healthier lives. The work you do...

Job Description

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're looking for a hands-on technical leader to design, build, and productionize AI/ML and GenAI solutions that improve healthcare operations and patient outcomes. You will own end-to-end delivery-from problem framing and data pipelines to models, MLOps/LLMOps, and ongoing monitoring-while mentoring a small team and partnering with product, data engineering, and clinical/operations stakeholders.

Primary Responsibilities:

  • Lead architecture and delivery of ML/GenAI solutions (classification, forecasting, NLP, deep learning, LLM apps) at production scale
  • Build robust data/feature pipelines over large clinical/claims datasets; write efficient, well-tested Python (pandas/NumPy) and SQL
  • Develop and deploy LLM capabilities: prompt design, fine-tuning, RAG pipelines, vector indexing, and evaluation with guardrails
  • Establish MLOps/LLMOps best practices: CI/CD, model registry, experiment tracking, monitoring, drift detection, A/B testing, cost/perf optimization
  • Ensure data privacy and compliance (HIPAA/PHI handling, access controls, auditability) and champion model governance and Responsible AI
  • Translate business problems into technical roadmaps; communicate trade-offs and results to executives and non-technical partners
  • Mentor engineers and set engineering standards (code reviews, documentation, reliability, observability)
  • Lead architecture and delivery of ML/GenAI solutions (classification, forecasting, NLP, deep learning, LLM apps) at production scale
  • Build robust data/feature pipelines over large clinical/claims datasets; write efficient, well-tested Python (pandas/NumPy) and SQL
  • Develop and deploy LLM capabilities: prompt design, fine-tuning, RAG pipelines, vector indexing, and evaluation with guardrails
  • Establish MLOps/LLMOps best practices: CI/CD, model registry, experiment tracking, monitoring, drift detection, A/B testing, cost/perf optimization
  • Ensure data privacy and compliance (HIPAA/PHI handling, access controls, auditability) and champion model governance and Responsible AI
  • Translate business problems into technical roadmaps; communicate trade-offs and results to executives and non-technical partners
  • Mentor engineers and set engineering standards (code reviews, documentation, reliability, observability)
  • 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 regard 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:

  • Bachelors in Computer Science, Engineering, Math, or related field required; MS/PhD preferred or equivalent experience
  • 10+ years of professional experience in software/ML engineering, including 3+ years leading projects or teams
  • Hands-on GenAI experience: LLMs, embeddings, RAG, fine-tuning, evaluation; familiarity with Hugging Face and LangChain/LlamaIndex
  • Solid foundation in statistics and ML (hypothesis testing, experimental design, feature engineering, supervised/unsupervised methods)
  • Deep learning expertise (PyTorch or TensorFlow) and modern NLP (transformers)
  • Expert Python and pandas stack; ability to write vectorized, high-performance code; solid testing practices (pytest) and Git
  • Solid data engineering skills: SQL; experience with Spark/Dask and workflow orchestration (Airflow/Prefect)
  • Cloud proficiency (AWS/Azure/GCP) and containerization/orchestration (Docker/Kubernetes); CI/CD and IaC (Terraform) exposure
  • Proven excellent communication and stakeholder management skills
  • Preferred Qualifications:

  • Healthcare domain experience: claims, EHR/HL7/FHIR, coding (ICD/CPT), risk adjustment, quality measures, de-identification
  • Big data platforms (Databricks, Snowflake, BigQuery) and streaming (Kafka); lakehouse patterns
    MLOps stack: MLflow/SageMaker/Azure ML/Vertex; model monitoring/observability
  • Vector databases (FAISS, Pinecone, pgvector), knowledge graphs (Neo4j), and ontologies (UMLS/SNOMED)
  • Security/compliance frameworks (SOC 2, HITRUST)
  • Additional languages for performance or integration (Scala/Java/Go)
  • #Exetech

    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.

This listing expired on 15 Jun. Applications are no longer accepted.

Below are some other jobs we think you might be interested in.

  • Lead AI/ML Engineer
    • RingCentral
    • Bengaluru, KA, IN
    Job Description Job Description:\nWe are seeking an experienced Lead AI/ML Engineer with a strong background in Natural Language Understanding (NLU)...
    13 Jun
  • Lead AI/ML Engineer
    • Fission Labs
    • Hyderabad, TG, IN
    Job Description Role: Lead AI & ML/ Associate Architect -AI/ML\nExperience: 8+ years\nLocation: Hyderabad\nFission Labs , headquartered in Sunnyvale,...
    16 Jun
  • Lead AI ML Engineer
    • 3Pillar
    • Noida
    We build breakthrough software products that power digital businesses. We are an innovative product development partner whose solutions drive...
    12 Jun
  • AI ML Engineer - Lead
    • Armakuni
    • Ahmedabad, GJ, IN
    Job Description We are looking for a highly skilled Lead AI/ML Engineer with 6+ years of hands-on experience in designing, deploying, and scaling AI/ML...
    16 Jun
  • AI/ML Engineer- Lead
    • Crescens Inc.
    • Bangalore, Karnataka, IN
    Job Title: AI/ML Engineer- Lead Location: Bangalore, Karnataka (onsite) Total Experience: 7 to 9 years Mode of Interview: Face to Face (Mandatory)...
    26 May
  • AI/ML Engineer- Lead
    • Crescens Inc.
    • Bangalore,Karnataka,India
    Job Title: AI/ML Engineer- Lead Location: Bangalore, Karnataka (onsite) Total Experience: 7 to 9 years Mode of Interview: Face to Face (Mandatory)...
    27 May
  • AI/ML Engineer- Lead
    • National Staffing
    • Bangalore,Karnataka,India
    Job Title: AI/ML Engineer- Lead Location: Bangalore, Karnataka (onsite) Total Experience: 7 to 9 years Mode of Interview: Face to Face (Mandatory)...
    27 May
  • AI / ML Engineer/Lead
    • Codvo.ai
    • Pune, Maharashtra, IN
    At Codvo, software and people transformations go hand-in-hand. We are a global empathy-led technology services company where product innovation and...
    12 Jun
  • AI / ML Engineer/Lead
    • Codvo.ai
    • Maharashtra,Pune,India
    At Codvo, software and people transformations go hand-in-hand. We are a global empathy-led technology services company where product innovation and...
    12 Jun
  • Lead ML & AI Engineer
    • HERE Technologies
    • Mumbai,IN,400063
    What's the role?The HERE NDS Technologies team is looking for a Lead AI/ML Engineer to drive NDS mapping systems. You will lead the design and...
    12 Jun
  • Lead AI/ML
    • TekFortune
    • Indore,Madhya Pradesh,452001
    What will your role look like Lead the design, development, and deployment of AI and Generative AI solutions using AWS, Azure, or GCP. Provide...
    12 Jun
  • Lead AI/ML
    • Tekfortune India IT Pvt Ltd.
    • Indore, Madhya Pradesh, IN
    What will your role look like Lead the design, development, and deployment of AI and Generative AI solutions using AWS, Azure, or GCP. Provide...
    12 Jun
  • AI/ML Lead
    • Open Financial Technologies
    • Bengaluru, KA, IN
    Job Description About Open: OPEN is a leading connected finance platform that empowers finance teams to manage their cash flow better by integrating...
    16 Jun
  • AI/ML Engineer
    • NR Consulting - India
    • Bangalore, Karnataka, IN
    Title: AI/ML Engineer Location: Bangalore Exp: 4 to 6 Years Job Description: " 3 8 years of experience in software development, AI/ML...
    27 May
  • AI/ ML Engineer
    • MaieuticSemi
    • Bangalore Urban, Karnataka, IN
    AI/ ML Engineer Position Overview We are seeking a skilled AI/ML Engineer with expertise in developing and deploying classical machine learning models...
    15 Jun
  • AI/ML Engineer
    • NR Consulting
    • Bangalore,Karnataka
    Title: AI/ML Engineer Location: Bangalore Exp: 4 to 6 Years Job Description: 3-8 years of experience in software development, AI/ML...
    29 May
  • AI/ML Engineer
    • Mindfire Solutions
    • Dombivali, MH, IN
    Job Description About the Job\nAs an AI/ML Engineer, you will be responsible for architecting, developing, and deploying production-grade Generative AI...
    16 Jun
  • AI/ML Engineer
    • Mindfire Solutions
    • Malappuram, KL, IN
    Job Description About the Job As an AI/ML Engineer, you will be responsible for architecting, developing, and deploying production-grade Generative AI...
    13 Jun
  • AI/ML Engineer
    • Mindfire Solutions
    • Panipat, HR, IN
    Job Description About the Job As an AI/ML Engineer, you will be responsible for architecting, developing, and deploying production-grade Generative AI...
    17 Jun
  • AI/ML Engineer
    • Mindfire Solutions
    • Thāne, MH, IN
    Job Description About the Job As an AI/ML Engineer, you will be responsible for architecting, developing, and deploying production-grade Generative AI...
    17 Jun