Associate AI/ML Engineer
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.
We are seeking a highly skilled Software Engineer (with 4-6 years of experience) with expertise in AI/ML and Salesforce development. This role focuses on designing and enhancing Salesforce applications by integrating advanced AI solutions, including Einstein GPT and custom models using Python or cloud platforms. Key responsibilities include building RAG pipelines, implementing real-time automation, and collaborating across teams to deliver scalable, secure, and high-performing solutions. This is a great opportunity for someone passionate about advancing enterprise AI beyond traditional tools.
Primary Responsibilities:
- Design, develop, test, document, and deploy Salesforce solutions based on business needs
- Develop and deploy AI/ML models for real-time decision-making and automation
- Integrate AI/ML solutions into Salesforce CRM to enable intelligent data retrieval, personalized recommendations, workflow automation, forecasting, scoring, and opportunity insights
- Enhance Salesforce applications with advanced AI features using both native (Einstein/Einstein GPT) and external technologies (Python-based models or Azure/AWS ML services)
- Build and optimize Retrieval-Augmented Generation (RAG) pipelines using vector databases and Large Language Models (LLMs) for improved contextual understanding within Salesforce workflows.
- Extend platform functionality using Apex (Triggers/Classes), LWC, Aura Framework, Visualforce Pages, Apex APIs and web services
- Create custom AI-powered automation solutions including Einstein Bots and bespoke bots for sales/service processes
- Manage CI/CD processes with Git to ensure efficient deployment of updates
- Collaborate with product managers, engineers, and data teams to identify opportunities for impactful AI-driven enhancements in the Salesforce ecosystem
- Ensure scalability, security/compliance (especially regarding sensitive customer data), reliability and performance of deployed models in production environments
- Monitor model accuracy and relevance; continuously improve user experience through feedback loops
- Stay current with advancements in generative AI (including transformers), enterprise-grade integrations/platforms/tools; evaluate applicability to organizational needs
- 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 regard 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
Want more jobs like this?
Get jobs in Gurgaon, India delivered to your inbox every week.

Required Qualifications:
- Salesforce ADM 201 and PD1 certifications
- Bachelor's or Master's degree in Computer Science, Data Science, AI/ML, or a related field
- 4+ years of hands-on experience in Salesforce development and AI/ML engineering, with a focus on NLP and generative models
- Proven experience with RAG architectures, vector databases (e.g., FAISS, Pinecone), and large language models (LLMs) such as GPT or LLaMA
- Experience establishing and supporting Salesforce Communities (Customer Community)
- Solid proficiency in Python and ML frameworks (e.g., PyTorch, TensorFlow, Hugging Face)
- Demonstrated ability to integrate AI systems with Salesforce CRM using APIs, Apex, or MuleSoft
- Solid background in Agile methodologies; experience working with distributed/offshore teams is a plus
- Familiarity with MLOps tools and practices (e.g., MLflow, Docker, Kubernetes)
Proven excellent communication skills; solid problem-solving abilities; ability to work both independently and collaboratively within multidisciplinary teams
Preferred Qualifications:
- Salesforce App Builder, Salesforce PD2 certification
- Experience with cloud platforms (AWS, Azure, GCP) and scalable AI deployments
- Experience in enterprise domains such as healthcare, finance, or customer service
- Knowledge of Salesforce data models, workflows, and automation tools
- Understanding of data privacy, compliance, and ethical AI principles
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.
Perks and Benefits
Health and Wellness
- HSA
- On-Site Gym
- HSA With Employer Contribution
- Health Insurance
- Dental Insurance
- Vision Insurance
- Life Insurance
- Short-Term Disability
- Long-Term Disability
- FSA
Parental Benefits
- Non-Birth Parent or Paternity Leave
- Birth Parent or Maternity Leave
Work Flexibility
- Remote Work Opportunities
Office Life and Perks
- Commuter Benefits Program
Vacation and Time Off
- Leave of Absence
- Personal/Sick Days
- Paid Holidays
- Paid Vacation
Financial and Retirement
- Relocation Assistance
- Performance Bonus
- Stock Purchase Program
- 401(K) With Company Matching
- 401(K)
Professional Development
- Promote From Within
- Shadowing Opportunities
- Access to Online Courses
- Tuition Reimbursement
Diversity and Inclusion
- Diversity, Equity, and Inclusion Program
Company Videos
Hear directly from employees about what it is like to work at UnitedHealth Group.