Senior AI/ML Engineer

UnitedHealth Group; Gurugram District; On-site; Confidential; Full time

Job summary

UnitedHealth Group seeks a software engineer with 5+ years in software engineering and 1–2 years in generative AI/LLMs to design, implement, and optimize AI-powered features. The role involves building cloud-based applications on Azure/AWS/GCP, leveraging FAISS/Pinecone/pgvector for vector DBs, and applying prompt engineering with RAG.

You will collaborate across product, data, and engineering teams to deliver scalable AI solutions while focusing on safety, evaluation, and performance.

Qualifications

• Bachelor's or master's degree in computer science, engineering, or related field.
• 5+ years of software engineering experience, including 1-2 years with generative AI or LLMs.
• Experience building cloud-based applications (Azure, AWS, or GCP).
• Knowledge of LLMs, prompt engineering, fine tuning basics, and RAG.
• Familiarity with transformer architectures, embeddings, evaluation and safety concepts.
• Experience integrating AI via APIs, vector databases, and orchestration frameworks.
• Proficiency in Python or TypeScript.

Responsibilities

• AI Applications and Feature Development: build and refine LLM-based features like chat, summarization, and data extraction.
• Model and Architecture Understanding: work with transformer models and embeddings.
• AI Quality, Evaluation and Safety: improve accuracy, reduce hallucinations, and run evaluations.
• Integration and Orchestration: connect AI APIs and implement vector DB workflows.
• Performance and Optimization: improve latency, throughput, and batching.
• Collaboration and Delivery: work with product, data, and engineering teams; participate in reviews.

Skills

LLMs

Prompt engineering

RAG

Transformers

APIs integration

Vector DBs

Python

TypeScript

Latency optimization

Problem solving

Collaboration

Education

Bachelor's or master's degree in Computer Science or Engineering

Tools

Azure

AWS

GCP

FAISS

Pinecone

pgvector

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:

• AI Applications and Feature Development
  • Develop and refine LLM-based features such as chat, summarization, knowledge search, and data extraction
  • Apply prompt engineering techniques and support model fine tuning or parameter efficient tuning (e.g., LoRA)
  • Implement RAG (retrieval augmented generation) workflows including preprocessing, embedding generation, and vector search
• Model and Architecture Understanding
  • Work with transformer-based models; apply foundational understanding of attention, embeddings, and model behavior
  • Assist in model interpretability and improving reliability of model outputs
• AI Quality, Evaluation and Safety
  • Help improve accuracy by reducing hallucinations through prompting, guardrails, and retrieval improvements
  • Support benchmarking, evaluation tests, and quality checks
  • Contribute to Responsible AI practices
• Integration and Orchestration
  • Build services integrating AI APIs (OpenAI, Azure, open source LLMs)
  • Implement tool/agent orchestration flows with guidance from senior team members
  • Work with vector databases such as FAISS, Pinecone, or pgvector
• Performance and Optimization
  • Contribute to enhancements in latency, throughput, and token efficiency
  • Support implementation of model compression and quantization techniques
  • Develop caching and batching strategies to improve system performance
• Collaboration and Delivery
  • Work closely with product, data, and engineering teams to deliver high quality AI features

  • Participate in code reviews, team design discussions, and knowledge sharing

• 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

Builder Responsibilities:

Design, develop, and deploy AI-powered solutions using no-code, low-code, and advanced platforms, translating business needs into scalable applications that enhance products, workflows, and decision-making.

Required Qualifications:

• Bachelor's or master's degree in computer science, Engineering, or a related field
• 5+ years of software engineering experience, including 1-2 years with generative AI or LLMs
• Experience building cloud-based applications (Azure, AWS, or GCP)
• Working knowledge of:
  • LLMs, prompt engineering, fine tuning basics, and RAG
  • Transformer architecture fundamentals (attention, embeddings)
  • Evaluation and safety concepts (hallucination reduction, quality checks)
  • AI integration using APIs, vector DBs, and orchestration frameworks
  • Token optimization and latency reduction strategies
• Proficiency in Python or TypeScript
• Proven solid problem solving and collaborative skills

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.