AI/ML Engineer

UnitedHealth Group; Bengaluru; On-site; INR 3,500,000 - 6,000,000; Full time

Job summary

Optum, part of the UnitedHealth Group family, invites you to join a team advancing health care through AI research and production systems.

Lead and mentor in AIML, collaborate with data scientists and engineers to deploy scalable models and governance practices, shaping solutions in health tech. This role emphasizes research excellence, responsible AI, and impact across health outcomes.

Qualifications

• Bachelors in CS, ML, or related field.
• 4+ years AIML engineering/research.
• Proven expertise in LLMs, generative AI.
• Strong Python and ML libraries knowledge.
• Experience with experiment tracking tools.
• Hands-on MLOps, deployment, monitoring.
• Familiar with AIML governance and ethics.

Responsibilities

• Lead applied research in LLMs, generative AI, and multimodal models.
• Evaluate architectures (Transformers, Diffusion, Retrieval-Augmented Generation) and publish internal papers.
• Design scalable AIML pipelines and deploy models in production.
• Collaborate with cross-functional teams and mentor juniors.
• Architect end-to-end AIML systems and align with business goals.

Skills

LLMs

Generative AI

Deep learning

Python

ML libraries

Experiment tracking

MLOps

Governance & ethics

Education

Bachelors Degree in Computer Science, Machine Learning, or related field

Tools

PyTorch

TensorFlow

Hugging Face

MLflow

GitHub Copilot

Windsurf

Vertex AI

Azure OpenAI

Hugging Face Transformers

Job description

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

Research & Development

• Lead applied research in LLMs, generative AI, and multimodal models.
• Evaluate and experiment with state-of-the-art architectures (e.g., Transformers, Diffusion Models, Retrieval-Augmented Generation).
• Publish internal whitepapers and contribute to external conferences where applicable.

Engineering & Implementation

• Design and implement scalable AIML pipelines using frameworks like PyTorch, TensorFlow, Hugging Face, and MLflow.
• Collaborate with engineering teams to deploy models into production using MLOps best practices (CI/CD, model versioning, monitoring).

Tooling & Infrastructure

• Evaluate and integrate advanced AIML tools such as GitHub Copilot, Windsurf, Vertex AI, Azure OpenAI, and Hugging Face Transformers.
• Work with cloud platforms (AWS, Azure, GCP) to ensure scalable and secure model deployment.

Architecture & Strategy

• Architect end-to-end AIML systems including data ingestion, model training, inference, and feedback loops.
• Partner with enterprise architects and product leaders to align AIML capabilities with business goals 3.

Mentorship & Collaboration

• Mentor junior engineers and researchers.
• Collaborate with cross-functional teams including data scientists, software engineers, and business stakeholders.

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.

Qualifications

• Bachelors Degree in Computer Science, Machine Learning, or related field.
• 4+ years of experience in AIML engineering and research.
• Proven expertise in LLMs, generative AI, and deep learning.
• Strong programming skills in Python and familiarity with ML libraries (e.g., scikit-learn, Keras).
• Experience with experiment tracking tools (e.g., Weights & Biases, MLflow).
• Hands-on experience with MLOps, model deployment, and monitoring.
• Familiarity with AIML governance, ethics, and responsible AI practices.

Careers with Optum. Here's the idea. We built an entire organization around one giant objective; make the health system work better for everyone. So when it comes to how we use the world's large accumulation of health-related information, or guide health and lifestyle choices or manage pharmacy benefits for millions, our first goal is to leap beyond the status quo and uncover new ways to serve. Optum, part of the UnitedHealth Group family of businesses, brings together some of the greatest minds and most advanced ideas on where health care has to go in order to reach its fullest potential. For you, that means working on high performance teams against sophisticated challenges that matter. Optum, incredible ideas in one incredible company and a singular opportunity to do your life's best work. SM

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.

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.