Data Engineering Scientist with AI/ML - Remote

Optum Tech is a global leader in health care innovation. Our teams develop cutting-edge solutions that help people live healthier lives and help make the health system work better for everyone. From advanced data analytics and AI to cybersecurity, we use innovative approaches to solve some of health care's most complex challenges. Your contributions here have the potential to change lives. Ready to build the next breakthrough? Join us to start Caring. Connecting. Growing together.

UHC Technology is focused on driving change, modernization, and ensuring reliable and stable systems so that we can help to transform health care - making it easier, more affordable, and more effective for those we serve. We are passionate about technology and the role it plays to create distinctive experiences to our constituents. While we are always focused on how technology can help us deliver faster and with improved quality, we are far more enthusiastic about the ways technology can reinvent how we deliver on our mission in partnership with the UHC lines of business and Optum Technology.

We are seeking a highly skilled and motivated AI/ML Engineer to lead innovation in claims adjudication through advanced Generative AI solutions. This role emphasizes Large Language Models (LLMs), agentic frameworks, and prompt engineering to automate complex workflows. You will design and deploy secure, scalable, and responsible AI systems while collaborating across teams to deliver measurable impact.

You'll enjoy the flexibility to work remotely * from anywhere within the U.S. as you take on some tough challenges. For all hires in the Minneapolis or Washington, D.C. area, you will be required to work in the office a minimum of four days per week.

Primary Responsibilities:

• Design, develop, and deploy AI/ML and Generative AI models for predictive, prescriptive, and generative analytics across healthcare datasets
• Implement advanced architectures including LLMs (GPT, Gemini, LLaMA), Retrieval-Augmented Generation (RAG), and Agentic Frameworks
• Build and optimize end-to-end pipelines using Python (Sci-kit Learn, Pandas, Flask, LangChain), PySpark, T-SQL and SQL
• Develop and fine-tune multiple GenAI models for NLP, summarization, prompt engineering, and conversational AI
• Apply MLOps best practices: model versioning, drift analysis, quantization, MLFlow, containerization with Docker, and CI/CD pipelines
• Work with cloud platforms: Azure (Databricks, ML Studio, Data Factory, Data Lake, Delta Tables), AWS, and Google Cloud Platform for scalable deployments
• Integrate data warehousing solutions like Snowflake and manage large-scale data pipelines.
• Collaborate in an Agile environment, participate in sprint planning, and maintain code repositories using GitHub/Git
• Ensure compliance with security and governance standards for healthcare data
• Coach and mentor junior team members

Technical Skillset

AI/ML Foundations

• Design and implement machine learning and deep learning models for classification, NLP tasks
• Build and maintain end-to-end ML pipelines including data preprocessing, model training, evaluation, and deployment

Generative AI & LLM Engineering

• Develop and fine-tune LLM-based applications using LangChain, LangGraph, and other GenAI frameworks
• Build Multi Agentic workflows and RAG (Retrieval-Augmented Generation) pipelines for enterprise use cases
• Leverage AWS Bedrock and Google Vertex AI for scalable and production-grade GenAI deployments

LLM Security & Responsible AI

• Implement guardrails to prevent prompt injection, reduce hallucinations, and ensure safe model outputs
• Apply best practices for LLM security, including output moderation, access control, and auditability
• Ensure compliance with Responsible AI principles-fairness, transparency, and explainability

Cloud-Native AI Development

• Deploy and manage GenAI solutions on AWS and Google Suite, utilizing services like Bedrock, SageMaker, Vertex AI
• Integrate LLMs with enterprise systems using REST APIs, SDKs, and orchestration tools

Collaboration & Mentorship

• Work closely with product managers, data scientists, and platform teams to translate business needs into GenAI solutions
• Mentor junior engineers and contribute to internal knowledge-sharing initiatives

You'll be rewarded and recognized for your performance in an environment that will challenge you and give you clear direction on what it takes to succeed in your role as well as provide development for other roles you may be interested in.

Required Qualifications:

• 5+ years of hands-on experience in AI/ML techniques like Prompt Engineering , RAG (Retrieval Augmented Generation) and Agentic AI
• Solid expertise in Python, PySpark, T-SQL, SQL, and big data technologies (Hadoop, Spark)
• Proven deep knowledge of statistics, data modeling, and simulation
• Hands-on experience with Generative AI frameworks/architectures (LangChain, HuggingFace, OpenAI APIs)
• Proficiency in cloud technologies: Azure (Databricks, ML Studio), AWS Bedrock, Azure Foundry, Kafka, and cloud-native AI services
• Demonstrated familiarity with CI/CD pipelines, GitHub Actions, and containerization tools
• Proven excellent problem-solving skills and ability to handle ambiguity
• Proven solid understanding of LLM security, prompt engineering, and responsible AI practices

Preferred Qualifications:

• Experience with LLMs (GPT, Gemini, LLaMA) and prompt-based learning
• Knowledge of Kafka, TensorFlow, and advanced deep learning architectures (CNNs, Autoencoders)
• Solid understanding of Agile methodologies and DevOps practices
• Internal Data management and Big data handling experience

*All employees working remotely will be required to adhere to UnitedHealth Group's Telecommuter Policy.

Pay is based on several factors including but not limited to local labor markets, education, work experience, certifications, etc. In addition to your salary, we offer benefits such as, a comprehensive benefits package, incentive and recognition programs, equity stock purchase and 401k contribution (all benefits are subject to eligibility requirements). No matter where or when you begin a career with us, you'll find a far-reaching choice of benefits and incentives. The salary for this role will range from $71,200 to $127,200 annually based on full-time employment. We comply with all minimum wage laws as applicable.

Application Deadline: This will be posted for a minimum of 2 business days or until a sufficient candidate pool has been collected. Job posting may come down early due to volume of applicants.

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 an Equal Employment Opportunity employer under applicable law and qualified applicants will receive consideration for employment without regard to race, national origin, religion, age, color, sex, sexual orientation, gender identity, disability, or protected veteran status, or any other characteristic protected by local, state, or federal laws, rules, or regulations.

UnitedHealth Group is a drug - free workplace. Candidates are required to pass a drug test before beginning employment.