Associate AI/ML Engineer

Taleo; Bengaluru; On-site; INR 1,500,000 - 2,500,000; Full time

Job summary

UnitedHealth Group in Bengaluru is seeking an experienced AI/ML engineer to design and develop the next generation of NLP/AI products for healthcare. You will build models across NLP, NLU, NLG, SLU, and time series forecasting, leveraging RAG, LangChain, and VectorDBs.

The role emphasizes end-to-end ML development, model optimization, and mentoring junior engineers, with strong collaboration across teams and a focus on healthcare impact.

Qualifications

• Bachelor’s or Master’s degree in Computer Science, Data Science, Statistics, or a related field.
• 3+ years of experience in AI/ML.
• Solid programming skills in Python, with pandas, numpy, and SQL.
• Experience with the full life cycle of AI/ML projects, including EDA and drift analysis.
• Proficiency in frameworks such as Scikit-Learn, TensorFlow, and PyTorch.

Responsibilities

• Help design and develop the next generation of NLP, ML & AI products for healthcare.
• Develop ML models and systems in NLP, NLU, NLG, SLU and time series forecasting.
• Work with RAG, LangChain, and VectorDBs to enable retrieval-augmented approaches.
• Quantize and optimize GenAI models and manage ML model lifecycles.
• Lead end-to-end ML development from data ingest to deployment and monitoring.
• Mentor junior engineers and stay updated with AI/ML research.

Skills

Python

Pandas

NumPy

SQL

Flask API

FastAPI

Scikit-Learn

TensorFlow

PyTorch

NLP

LangChain

RAG

VectorDBs

Model lifecycle

ML/DL theory

Education

Bachelors in CS

Masters in CS

DS/Statistics background

Tools

TensorFlow

PyTorch

Scikit-Learn

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 Responsibilities:

• Help design and develop the next generation of NLP, ML & AI products, and services for healthcare


• Develop machine learning and deep learning models and systems in domains including, but not limited to: NLP, NLU, NLG, SLU and multidimensional time series forecasting among others


• Exposure in RAG, LangChain, VectorDBs


• Ability to Quantize, Optimize GenAI models


• Manage NLP & ML models lifecycle for a suite of products


• Run large complex proof-of-concepts for the healthcare business


• Manage prioritization and technology work for building NLP, ML & AI solutions


• Lead the full end-to-end machine learning development process including data ingestion and preparation, feature engineering, analysis and modeling, model deployment, performance tracking and documentation


• Establish best practices for end-to-end deep learning and machine learning development cycle to ensure rigor in process and quality in outcome


• Work with a great deal of autonomy to find solutions to complex problems


• Design and architect scalable ML solutions for complex business problems


• Lead Projects: Lead ML projects from conception to deployment, ensuring timely delivery and quality


• Mentorship: Mentor junior engineers and provide technical guidance to the team


• Research and Innovation: Stay updated with the latest research in AI/ML and implement cutting-edge techniques.


• Cross-functional Collaboration: Collaborate with cross-functional teams to align ML solutions with business goals


• Comply with all applicable Company policies, procedures, and business directives, changes including those relating to work location, team assignments, work schedules, and flexible work arrangements

Required Qualifications:

• Bachelor’s or Master’s degree in Computer Science, Data Science, Statistics, or a related field


• 3+ years of experience in AI/ML


• Solid programming skills in Python, with experience in pandas, numpy, and SQL


• Experience with the full life cycle of AI/ML projects, including EDA, model development, tuning, and drift analysis


• Solid knowledge of discriminative AI model development and generative AI applications


• Familiarity with Python frameworks such as Flask API and FastAPI
  Math behind ML and DL algorithms


• Proficiency in frameworks such as Scikit Learn, TensorFlow, and PyTorch

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.