Senior AI/ML Engineer

Optum; Bengaluru; On-site; INR 4,000,000 - 8,000,000; Full time

Job summary

Optum is a global health tech leader applying technology to improve health outcomes. This role focuses on enterprise-scale AI and optimization, translating business challenges into actionable AI/ML strategies and production-ready architectures.

Candidates should own end-to-end optimization initiatives, work with engineering teams to deploy solutions, and apply hybrid ML/optimization approaches in healthcare analytics and decision systems.

Qualifications

• Bachelor's degree in a relevant field.
• 10+ years of experience in applied AI/ML or optimization roles with ownership of enterprise-scale AI/optimization initiatives.
• Proven experience translating business challenges into optimization/AI solution strategies and production-ready architectures.
• Hands-on experience with optimization frameworks such as Pyomo, OR-Tools, Gurobi, and/or CPLEX.
• Hands-on experience with ML/DL frameworks like PyTorch and/or TensorFlow.

Responsibilities

• Comply with employment terms, policies, and directives as required by the company.

Skills

AI optimization

Python

SQL

PyTorch

TensorFlow

Education

Bachelor's degree

Tools

Pyomo

OR-Tools

Gurobi

CPLEX

Qiskit

Job description

Improve the lives of others while Caring. Connecting. Growing together.

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 Responsibility

• 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

• Bachelor's degree in Computer Science, Mathematics, Engineering, Operations Research, Applied Mathematics, or related field
• 10+ years of experience in applied AI/ML or optimization-focused roles, with solid ownership of enterprise‑scale AI and optimization initiatives
• Proven experience translating business challenges into effective optimization and AI/ML solution strategies and production‑ready solution architectures
• Hands‑on experience with optimization frameworks, including Pyomo, OR-Tools, Gurobi, and/or CPLEX
• Hands‑on experience with ML/DL frameworks: PyTorch and/or TensorFlow
• Experience with combinatorial optimization and large‑scale decision systems
• Demonstrated experience with hybrid ML and optimization approaches
• Solid foundation in:
• Mathematical optimization, operations research, and statistics
• Machine learning, deep learning, and experimental design
• Demonstrated experience designing, developing, validating, and delivering successful optimization and AI proof‑of‑concepts that progressed into large‑scale production environments
• Familiarity with quantum computing concepts or quantum‑inspired algorithms for optimization
• Familiarity with MLOps practices including experiment tracking, benchmarking, model versioning, and deployment automation
• Proven ability to translate theoretical models into impactful, real‑world solutions
• Proven ability to collaborate effectively with engineering organizations to enable successful production adoption of AI and optimization solutions
• Proven ability to collaborate effectively across business, product, engineering, and leadership teams
• Proven solid programming skills in Python (NumPy, pandas, scientific computing) and SQL
• Proven solid analytical, problem‑solving, communication, solution design, and stakeholder management skills

Preferred Qualifications

• MS/PhD
• Experience with quantum frameworks such as Qiskit, Cirq, D-Wave, or Azure Quantum
• Experience with reinforcement learning for decision optimization
• Experience with simulation‑based optimization and digital twins
• Experience integrating optimization into production systems in collaboration with engineering teams
• Experience building enterprise‑scale Generative AI solutions integrated with optimization pipelines
• Experience mentoring teams and driving AI capability development across organizations
• Experience with Big data platforms (Databricks, Snowflake, BigQuery) and streaming (Kafka); lakehouse patterns
• Experience with MLOps stack: MLflow, SageMaker, Azure ML, or Vertex AI; model monitoring, observability, automated retraining, and deployment automation
• Experience with Publications or patents in optimization, ML, or quantum computing; contributions to internal frameworks, accelerators, or enterprise AI innovation initiatives
• Healthcare domain expertise including claims, EHR/HL7/FHIR, care pathways, resource planning, ICD/CPT coding, risk adjustment, quality measures, and de‑identification
• Knowledge of Responsible AI in constrained decision‑making systems
• Knowledge of security and compliance frameworks (SOC 2, HITRUST, HIPAA)

Technical Skills

• Optimization - Mathematical Optimization, Linear/Non-linear Programming, Integer and Combinatorial Optimisation, Stochastic Optimization, Robust Optimization, Operations Research, Large-Scale Decision Systems
• Quantum and Quantum‑Inspired - Quantum Optimisation, QAOA, Quantum Annealing, Hybrid Quantum-Classical Algorithms, Qiskit, Cirq, D-Wave, Azure Quantum
• AI/ML and Analytics - Machine Learning, Deep Learning, Statistical Modelling, Predictive Analytics, Prescriptive Analytics, Experimental Design
• Generative and Agentic AI - Generative AI, Large Language Models (LLMs), Retrieval‑Augmented Generation (RAG), Prompt Engineering, Retrieval Optimisation, Constrained Generation
• AI Frameworks - PyTorch, TensorFlow
• Programming and Data Engineering - Python (NumPy, pandas, SciPy), SQL, Feature Engineering, Data Pipelines, Model Evaluation, Experimentation Frameworks
• MLOps and Model Lifecycle - MLOps, MLflow, Kubeflow, CI/CD, Model Monitoring, Observability, Drift Detection, Model Registry, Deployment Automation
• Data Platforms - Databricks, Snowflake, BigQuery, Kafka, Lakehouse Architectures
• Responsible AI and Governance - Responsible AI, Explainability, Fairness Constraints, AI Governance, Risk Management, SOC 2, HITRUST, HIPAA
• Healthcare Analytics - Healthcare Analytics, Claims, Clinical Data, EHR/HL7/FHIR, ICD/CPT, Risk Adjustment, Population Health, Care Management
• Research and Innovation - Operations Research, Quantum Computing, White Papers, Patents, Internal Frameworks, Innovation Leadership
• Leadership and Strategy - Technical Leadership, AI Strategy, Optimisation Strategy, Innovation, Stakeholder Management, Mentoring, Cross‑Functional Collaboration

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.

UnitedHealth Group is committed to working with and providing reasonable accommodations to individuals with physical and mental disabilities. If you need special assistance or accommodation for any part of the application process, please call 1-866-566-8715 to be connected to Recruitment Services. Recruitment Services hours of operation are 7 a.m. to 7 p.m. CT, Monday through Friday.

UnitedHealth Group is a registered service mark of UnitedHealth Group, Inc. The UnitedHealth Group name with the dimensional logo, as well as the dimensional logo alone, are both service marks for the UnitedHealth Group, Inc.

Diversity creates a healthier atmosphere: UnitedHealth Group is an Equal Employment Opportunity/Affirmative Action employer and all qualified applicants will receive consideration for employment without regard to race, color, religion, sex, age, national origin, protected veteran status, disability status, sexual orientation, gender identity or expression, marital status, genetic information, or any other characteristic protected by law.

UnitedHealth Group is a drug‑free workplace. Candidates are required to pass a drug test before beginning employment.