Senior AI/ML Engineer

Optum India; Gurugram District; On-site; INR 4,200,000 - 7,000,000; Full time

Job summary

Optum India is seeking a senior AI/ML software engineer to design, develop, and deploy production-grade machine learning models and generative AI solutions. You will implement LLMs, RAG systems, and transformer-based architectures, while deploying robust CI/CD pipelines and cloud-based workflows in Azure and AWS.

You will collaborate with cross-functional teams to preprocess large datasets, ensure security and governance for AI systems, and translate business needs into scalable AI/ML solutions.

Qualifications

• 8+ years of total software engineering and AI/ML experience.
• 3+ years hands-on experience developing and deploying AI-based projects in production environments.
• Proven expertise in multiple ML approaches and techniques.
• Track record of containerizing and deploying AI/ML projects at scale.
• Experience building and maintaining CI/CD pipelines for ML workflows.

Responsibilities

• Design, develop, and deploy ML models for production environments using supervised, unsupervised, RL, deep learning, and NLP.
• Build and optimize generative AI solutions including LLMs, RAG, transformers, and agentic AI systems.
• Deploy, fine-tune, and integrate LLMs from providers like OpenAI with prompt engineering and optimization.
• Conduct model testing and performance optimization with rigorous evaluation metrics.
• Design containerized AI/ML solutions using Docker and Kubernetes with CI/CD pipelines (GitHub Actions).
• Leverage Azure and AWS services for AI/ML projects including Databricks, AKS/EKS, Data Factory, Cognitive Services, Logic Apps.
• Collaborate with cross-functional teams to preprocess and interpret large datasets, translating business requirements into scalable AI/ML solutions.
• Identify and resolve code vulnerabilities, enforce security and governance for responsible AI practices.
• Comply with company policies and flexible work arrangements.

Education

Undergraduate degree

Tools

LangGraph

LangChain

OpenAI

Docker

Kubernetes

Azure

AWS

Databricks

AKS/EKS

Azure Data Factory

Cognitive Services

Logic Apps

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.

Primary Responsibilities

• Design, develop, and deploy sophisticated machine learning models for production environments using supervised, unsupervised, reinforcement learning, deep learning, and advanced NLP techniques
• Build and optimize generative AI solutions including Large Language Models (LLMs), Retrieval-Augmented Generation (RAG), transformer architectures, and agentic AI systems using frameworks like LangGraph and LangChain
• Deploy, fine-tune, and integrate LLMs from providers such as OpenAI, demonstrating expertise in prompt engineering, model customization, and optimization techniques
• Conduct comprehensive model testing, validation, and performance optimization using rigorous evaluation metrics to ensure production-grade quality and reliability
• Design and implement containerized AI/ML solutions using Docker and Kubernetes, establishing robust CI/CD pipelines with GitHub Actions for automated testing, validation, and deployment
• Leverage Azure and AWS services for AI/ML projects including Databricks, AKS/EKS, Azure Data Factory, Cognitive Services, and Logic Apps, managing system scaling and cloud migrations
• Collaborate with cross-functional teams to preprocess, analyze, and interpret extensive datasets; translate business requirements into scalable AI/ML solutions
• Identify and resolve code vulnerabilities, implement security best practices, and ensure compliance with AI/ML governance frameworks, ethics guidelines, and responsible AI practices
• 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

Required Qualifications

• Undergraduate degree or equivalent experience
• 8+ years of total software engineering and AI/ML experience
• 3+ years of hands‑on experience developing and deploying AI‑based projects in production environments
• Demonstrated expertise in multiple machine learning approaches and techniques
• Proven success containerizing and deploying AI/ML projects at scale
• Track record of implementing and maintaining CI/CD pipelines for ML workflows

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.

#NIC