AI ML Engineer
Optum is seeking a senior AI ML Engineer to advance health optimization through sophisticated AI solutions. The role involves architecting GraphRAG pipelines, training large language models, and building distributed training environments. The ideal candidate brings extensive experience in PyTorch, vector databases, and reinforcement learning. This position requires deep expertise in NLP and transformer-based architectures to build production-grade healthcare AI.
50k new jobs listed every day. Install TAL to find more jobs like this.

Experience
6+ years
Function
Engineering
Work mode
Onsite, India
Company
Tier 2
What you will work on
Optum is seeking a senior AI ML Engineer to advance health optimization through sophisticated AI solutions. The role involves architecting GraphRAG pipelines, training large language models, and building distributed training environments. The ideal candidate brings extensive experience in PyTorch, vector databases, and reinforcement learning. This position requires deep expertise in NLP and transformer-based architectures to build production-grade healthcare AI.
TAL's take
High-impact AI/ML role at a large-scale healthtech organization involving advanced generative AI and RAG architecture.
The JD provides a highly detailed list of specialized AI/ML technologies and architectural responsibilities.
Salaries at Optum
21.9 LPA average
Based on 181 Grapevine salary entries for Optum.
Engineering
0 - 2 years | 25
11 LPA average
Range: 7 - 15 LPA
Engineering
2 - 4 years | 25
15 LPA average
Range: 5 - 41 LPA
Engineering
4 - 6 years | 26
27 LPA average
Range: 8 - 157 LPA
Engineering
6 - 8 years | L1
20 LPA average
Range: 13 - 26 LPA
Must haves
- 6+ years experience in applied AI/ML with statistics
- PhD or Master's degree in computer science or related field
- Experience with PyTorch and LLM fine-tuning
- Experience with GraphRAG and vector databases
- Solid proficiency in Python and SQL
Tools and skills
About the company
Optum is a major healthcare technology company and a subsidiary of UnitedHealth Group, classified as a large enterprise.
Posts mentioning Optum
Please help me choose!
Between Optum, Skillsoft, Tech Mahindra which one would be a better choice in terms of work life balance, professional and financial growth? Optum : x LPA fixed SkillSoft : (x+2) LPA fixed Tech Mahindra : (x+3) LPA fixed
Looking for Referral at Optum
Hi Guys, I am looking for a job change. I am having the job I'd of Optum. Can anyone help me with the referral.
Optum RRP
How much rrp did you guys got ? Please share the value and performance score out of 5.