Senior AI/ML Engineer - Bangalore

Optum India; Bengaluru; On-site; INR 1,200,000 - 1,800,000; Full time

Job summary

Optum India in Bengaluru seeks an AI/ML Engineer to design and deploy machine learning and generative AI solutions across product platforms. You will manage data pipelines, train and evaluate models, and implement MLOps for production reliability.

The role emphasizes collaboration with data scientists, software engineers, and product managers to deliver AI-driven features at scale. The ideal candidate brings hands-on experience with ML frameworks, LLM workflows, cloud services, and strong Python

Qualifications

• Hands-on experience with ML frameworks and libraries such as TensorFlow, PyTorch, Keras, and Scikit-learn.
• Experience building LLM workflows and agentic applications using LangGraph and LangChain.
• Hands-on experience integrating LLM provider APIs (e.g., Anthropic Claude, OpenAI, Azure OpenAI) into applications.
• Deep understanding of machine learning, deep learning, and statistical modeling concepts.
• Familiarity with cloud platforms (AWS, Google Cloud, Azure) and their ML services.
• Familiarity with RAG, embeddings, and vector databases for retrieval-augmented generation.
• Proficiency in data handling (SQL) and libraries like Pandas and NumPy.
• Proficiency with Pydantic for data validation and structured outputs from LLMs.

Responsibilities

• Design and Develop ML Models to address business challenges from data to deployment.
• Build and maintain robust data pipelines for model training.
• Train, evaluate, and validate ML models to meet requirements.
• Deploy models into production and implement MLOps for lifecycle management.
• Collaborate with data scientists, engineers, and product managers to integrate AI features.
• Stay current with AI advancements and apply research to our products.
• Design and optimize LLM-powered workflows including RAG and multi-step orchestration.
• Establish evaluation frameworks and guardrails for safe, reliable generative AI.

Skills

TensorFlow

PyTorch

Keras

Scikit-learn

Python

SQL

Pandas

NumPy

Pydantic

LLM concepts

MLOps

Tools

LangGraph

LangChain

LLM APIs

Docker

Kubernetes

Databricks

Vector DBs

Embeddings

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 are seeking a talented and passionate AI/ML Engineer to join our innovative technology team. The ideal candidate will be at the heart of designing and building the intelligent systems that power our products and services. This role involves working on the complete lifecycle of machine learning projects, from data collection and preprocessing to model development, deployment, and maintenance. You will be a key player in transforming complex data into actionable insights and creating AI-driven solutions that solve real-world problems.

Primary Responsibilities

• Design and Develop ML Models: Research, design, and develop machine learning and deep learning models to address business challenges. This includes everything from selecting the right algorithms to building predictive models from the ground up


• Data Pipeline and Preprocessing: Build and maintain robust data pipelines. You will be responsible for gathering data from various sources, cleaning it, and transforming it into a usable format for model training


• Model Training and Evaluation: Train, test, and validate machine learning models to ensure they are accurate, reliable, and meet business requirements. This involves performing statistical analysis and running experiments to optimize model performance


• Deployment and MLOps: Deploy trained models into production environments and maintain their lifecycle. You will implement MLOps best practices for monitoring, retraining, and updating models to ensure they remain effective over time.


• Collaboration and Innovation: Work closely with data scientists, software engineers, and product managers to integrate AI/ML models into our applications and systems


• Stay Current: Keep up-to-date with the latest advancements in the field of AI, machine learning, and deep learning, and be prepared to apply new research to our products.


• Build Generative AI and LLM Workflows: Design, develop, and optimize LLM-powered applications and agentic workflows - including retrieval-augmented generation (RAG), prompt engineering, tool/function calling, and multi-step orchestration - to deliver production-grade generative AI features


• Ensure LLM Reliability and Evaluation: Establish evaluation frameworks, guardrails, and structured, type-safe output validation to ensure generative AI systems are accurate, safe, cost-efficient, and reliable in production


• 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

• Technical Skills:
  
  
  
  • Hands-on experience with ML frameworks and libraries such as TensorFlow, PyTorch, Keras, and Scikit-learn
  
  
  • Experience building LLM workflows and agentic applications using orchestration frameworks such as LangGraph and LangChain
  
  
  • Hands-on experience integrating LLM provider APIs (e.g., Anthropic Claude, OpenAI, Azure OpenAI) into applications
  
  
  • Deep understanding of machine learning, deep learning, and statistical modeling concepts
  
  
  • Familiarity with cloud platforms (e.g., AWS, Google Cloud, Azure) and their machine learning services (Databricks)
  
  
  • Familiarity with core generative AI concepts including retrieval-augmented generation (RAG), prompt engineering, embeddings, and vector databases (e.g., Pinecone, FAISS, Chroma)
  
  
  • Solid programming proficiency in languages like Python, R, or Java
  
  
  • Proficiency in data handling, including SQL, and experience with data processing tools like Pandas and NumPy
  
  
  • Proficiency with Pydantic for data validation and enforcing structured, type-safe outputs from LLMs
  
  
  
  


• Soft Skills:
  
  
  
  • Solid analytical and problem-solving abilities
  
  
  • Excellent communication skills to articulate complex technical concepts to diverse audiences
  
  
  • A creative and proactive mindset with a drive for continuous improvement
  
  
  
  

Preferred Qualifications

• Experience with containerization technologies like Docker and Kubernetes


• Proven experience deploying and managing machine learning models in a production environment


• Experience with LLM observability and evaluation tooling, fine-tuning, or prompt-optimization techniques


• Previous experience on databrick


• Experience with contributions to open-source projects or a solid portfolio of relevant personal projects


• Knowledge of MLOps principles and tools

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.