Skip to main content
Search JobsOpen search form

Explore remote jobs

Pursue your passion and potential

Lead AI or ML Engineer

Hyderabad, India

Caring. Connecting. Growing together.

With these values to guide us, our people are committed to making a meaningful difference in the lives of those we are honored to serve.

Lead AI or ML Engineer

Requisition number: 2370498 Job category: Technology Primary location: Hyderabad, Telangana Date posted: 06/15/2026 Overtime status: Exempt Travel: No

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:

  • AI Architecture & Solution Design:
    • Design enterprise-grade AI/ML and Generative AI architectures aligned with business and technology strategies
    • Define scalable patterns for: 
      • LLM integration
      • RAG (Retrieval-Augmented Generation)
      • Agentic AI workflows
      • AI orchestration frameworks
      • Vector databases
      • AI gateways
      • Model serving and inference
    • Architect multi-cloud AI solutions across Azure, AWS, and GCP
    • Define AI platform standards, reusable components, and reference architectures
  • Hands-on Engineering:
    • Build production-ready AI services and APIs using Python, Java, or Node.js.
    • Develop and optimize: 
      • Prompt engineering frameworks
      • AI agents
      • Semantic search pipelines
      • Embedding workflows
      • Fine-tuning pipelines
    • Implement integrations with enterprise systems using APIs, event-driven architectures, and streaming technologies
    • Work directly with engineering teams on coding, troubleshooting, performance tuning, and deployment automation
  • Generative AI & LLM Engineering:
    • Evaluate and implement enterprise LLM solutions including OpenAI, Anthropic, Gemini, and open-source models
    • Design secure and compliant GenAI architectures with guardrails, governance, and observability
    • Implement: 
      • RAG pipelines
      • Vector search
      • Knowledge grounding
      • AI evaluation frameworks
      • Hallucination mitigation strategies
    • Establish responsible AI practices including explainability, privacy, and bias mitigation
  • Cloud & Platform Engineering:
    • Design AI infrastructure on Azure OpenAI, AWS Bedrock, Vertex AI, Kubernetes, and container platforms
    • Build CI/CD pipelines for ML and AI workloads
    • Implement MLOps and LLMOps practices for model lifecycle management
    • Drive infrastructure automation using Terraform, Helm, GitHub Actions, or Azure DevOps
  • Data & Integration:
    • Collaborate with data engineering teams to build scalable AI data pipelines
    • Design event-driven and streaming architectures using Kafka and modern integration frameworks
    • Ensure interoperability and API-first design principles across platforms
  • Governance & Leadership:
    • Define enterprise AI standards, policies, and architecture governance
    • Mentor engineers and architects on AI best practices
    • Conduct architecture reviews and technology evaluations
    • Partner with security and compliance teams to ensure adherence to enterprise standards
  • 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


You'll be rewarded and recognized for your performance in an environment that will challenge you and give you clear direction on what it takes to succeed in your role as well as provide development for other roles you may be interested in.

Required Qualifications:

  • Bachelor's or Master's degree in Computer Science, Engineering, AI, Data Science, or related field
  • 10+ years of software engineering/architecture experience
  • Hands-on experience with AI/ML or Generative AI implementation
  • Experience deploying enterprise solutions in production environments
  • Solid expertise in: 
    • Python
    • APIs & microservices
    • Cloud-native architectures
    • Kubernetes & containers
    • AI/ML frameworks


Preferred Qualifications:

  • AI/Cloud certifications
  • Experience in healthcare, finance, or regulated industries
  • Experience with enterprise API governance and streaming platforms
  • Knowledge of responsible AI frameworks and AI security controls
  • Familiarity with FHIR, HL7, or healthcare interoperability standards
  • Proven solid communication and stakeholder management
  • Proven ability to influence executive and engineering leadership
  • Proven solid problem-solving and architecture decision-making skills
  • Proven ability to balance strategic thinking with hands-on execution


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.

Benefits

Our mission of helping people live healthier lives extends to our team members. Learn more about our range of benefits designed to help you live well.

Life

Resources and support to focus on what matters most to you, in every facet of your life.

Emotional

Education, tools and resources to help you reduce and manage stress, build resilience and more.

Physical

Health plans and other coverage to support wellness for you and your loved ones.

Financial

Benefits for today and to help you plan for the future, including your retirement.

Learn more
testimonial-img-1

Since joining Optum, my professional growth has been significant. The dynamic environment has enhanced my problem-solving abilities, and the company’s commitment to innovation and continuous learning motivates me to stay. Optum provides continuous training, mentorship, and a clear path for advancement, all while supporting a healthy work-life balance.

Anurag J.

Senior Software Engineering Manager

We’re honored to be recognized for our exceptional work culture

AGWF recognition award
2025 Campus Forward Award badge from RippleMatch
LinkedIn Top Companies 2025 award badge
Forbes Best Large Employers in the United States 2024 award badge
America’s Greatest Workplaces 2024 award badge