AI/ML Engineer

Posted 5 Hours Ago
Be an Early Applicant
Pune, Mahārāshtra, IND
In-Office
Mid level
Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
The Role
Design, build, evaluate, and operationalize AI/ML components (LLM workflows, RAG, agents, classifiers, summarizers). Ensure safety, evaluation, monitoring, and enterprise readiness while partnering with data, QA, security, and compliance teams.
Summary Generated by Built In
Requisition Number: 2371575
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.
The AI / ML / Agent Engineer owns the intelligence layer of the solution. This person designs, builds, evaluates, and improves the AI components, including LLM workflows, RAG pipelines, agentic behaviors, classifiers, summarizers, copilots, and AI evaluation frameworks.
This role ensures the AI is useful, measurable, safe, and fit for enterprise use.
Primary Responsibilities:
  • Design AI solution patterns aligned to the business problem and risk profile
  • Build LLM workflows, prompt chains, agents, RAG patterns, model integrations, classification models, summarization flows, or decision-support tools
  • Develop prompt strategies, tool-calling patterns, grounding techniques, guardrails, and fallback behaviors
  • Partner with Data Engineer to ensure the AI has access to trusted, appropriate, and governed data
  • Build and maintain evaluation datasets, scoring rubrics, model-performance tests, and failure-mode analysis
  • Monitor and improve output quality, hallucination risk, relevance, accuracy, consistency, latency, and cost
  • Partner with QA, Responsible AI, security, privacy, compliance, and architecture teams
  • Document model behavior, known limitations, risk controls, and operational monitoring needs
  • Support production readiness for AI components, including model observability and drift monitoring
  • 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:
  • Full-time graduation degree
  • Practical experience building AI-enabled applications, not just experiments
  • Experience with model APIs, orchestration frameworks, data pipelines, and evaluation tooling
  • Understanding of model limitations, hallucination risks, bias/fairness concerns, and responsible AI practices
  • Solid understanding of LLMs, RAG, embeddings, vector search, prompt engineering, agent frameworks, model evaluation, and AI safety
  • Solid Python or equivalent engineering skills
  • Ability to design AI workflows that are explainable, testable, and governable

Preferred Qualifications:
  • ML engineer, AI engineer, applied scientist, data scientist with engineering depth, or FDE with AI specialization
  • Experience building LLM applications in enterprise settings
  • Experience with controlled testing, model evaluation, and production AI monitoring

Success measures
  • AI outputs are relevant, accurate, useful, and measurable
  • Evaluation framework is in place before broad rollout
  • Known AI risks are identified and mitigated
  • AI performance improves through feedback and iteration
  • Solution passes Responsible AI, security, and compliance expectations
  • AI components are understandable and supportable by steady-state teams

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.

Skills Required

  • Full-time graduation degree
  • Practical experience building AI-enabled applications (production, not just experiments)
  • Experience with model APIs, orchestration frameworks, data pipelines, and evaluation tooling
  • Understanding of model limitations, hallucination risks, bias/fairness, and responsible AI practices
  • Solid understanding of LLMs, RAG, embeddings, vector search, prompt engineering, agent frameworks, model evaluation, and AI safety
  • Solid Python or equivalent engineering skills
  • Ability to design AI workflows that are explainable, testable, and governable
  • Experience as ML engineer, AI engineer, applied scientist, data scientist with engineering depth, or FDE with AI specialization
  • Experience building LLM applications in enterprise settings
  • Experience with controlled testing, model evaluation, and production AI monitoring

What the Team is Saying

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Eden Prairie, MN
160,000 Employees
Year Founded: 2011

What We Do

Optum, part of the UnitedHealth Group family of businesses, 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. At Optum, we support your well-being with an understanding team, extensive benefits and rewarding opportunities. By joining us, you’ll have the resources to drive system transformation while we help you take care of your future. We recognize the power of connection to drive change, improve efficiency and make a difference in health care. Join a team where your skills and ideas can make an impact and where collaboration is key to creating technology that produces healthier outcomes.

Gallery

Gallery
Gallery
Gallery

Optum Offices

Hybrid Workspace

Employees engage in a combination of remote and on-site work.

Optum has three workplace models that balance the needs of the business and the responsibilities of each role. These models, core on‑site (5 days/week), hybrid (4 days/week) and telecommute or fully remote, vary by country, role and location.

Typical time on-site: Not Specified
HQEden Prairie, MN
Metro Manila, Philippines
Cebu, Philippines
Davao, Philippines
Ann Arbor, MI
Atlanta, GA
Baltimore, MD
Bengaluru, India
Chennai, India
Dallas, TX
Detroit, MI
Dublin, Ireland
Hartford, CT
Houston, TX
Hyderabad, India
Jacksonville, FL
Las Vegas, NV
Letterkenny, Ireland
Louisville, KY
Madison, WI
Minneapolis, MN
Nashville, TN
New Delhi, India
Philadelphia, PA
Phoenix, AZ
Pune, India
Raleigh, NC
San Diego, CA
Washington, DC
Learn more

Similar Jobs

Optum Logo Optum

Machine Learning Engineer

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
In-Office
Pune, Mahārāshtra, IND
160000 Employees

Optum Logo Optum

Machine Learning Engineer

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
In-Office
Pune, Mahārāshtra, IND
160000 Employees

Optum Logo Optum

Machine Learning Engineer

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
In-Office
Pune, Mahārāshtra, IND
160000 Employees

Optum Logo Optum

Machine Learning Engineer

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
In-Office
Pune, Mahārāshtra, IND
160000 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account