Skip to main contentA logo with &quat;the muse&quat; in dark blue text.

Associate AI/ML Engineer

Yesterday Gurgaon, India

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.

Job alert subscription

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.

Client-provided location(s): Gurgaon, India
Job ID: UHG-2324157
Employment Type: FULL_TIME
Posted: 2025-11-13T20:38:15

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.