Senior AI/ML Engineer - Remote

UnitedHealth Group • Jersey City, NJ

1d • Remote

About The Position

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.
Optum AI is UnitedHealth Group’s enterprise AI team. We are AI/ML scientists and engineers with deep expertise in AI/ML engineering for health care. We develop AI/ML solutions for the highest impact opportunities across UnitedHealth Group businesses including UnitedHealthcare, Optum Financial, Optum Health, Optum Insight, and Optum Rx. In addition to transforming the health care journey through responsible AI/ML innovation, our charter also includes developing and supporting an enterprise AI/ML development platform.
Optum AI team members:
Have impact at scale: We have the data and resources to make an impact at scale. When our solutions are deployed, they have the potential to make health care system work better for everyone
Do ground-breaking work: Many of our current projects involve cutting edge ML, NLP and LLM techniques. Generative AI methods for working with structured and unstructured health care data are continuously being developed and improved. We are working in one of the most important frontiers of AI/ML research and development
Partner with world-class experts on innovative solutions: Our team members are developing novel AI/ML solutions to business challenges. In some cases, this includes the opportunity to file patents and publish papers about the methods we develop. We also collaborate with AI/ML researchers at some of the world’s top universities
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.
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.

Requirements

• Advanced degree in computer science, Machine Learning, or a related quantitative field
• 3+ years of experience in using ML to solve problems in natural language processing (NLP), or other AI domains
• 2+ years of experience in Generative AI, grounding architectures, prompt engineering, fine-tuning LLMs and/or improving pre-train LLM’s performance
• 2+ years of DL frameworks such as PyTorch and/or TensorFlow
• 2+ years of solid NLP research and developing experience
• Demonstrated ability to implement and update research-based state-of-the-art LLM methods

Nice To Haves

• Experience in end-to-end ML pipelines
• Experience with cloud computing, multi-threading, dynamic batch size adjustment
• Familiarity with advanced document parsing methods
• Familiarity with healthcare data (prior authorization, coverage documents, …)
• Proficiency with SaaS LLMs, including LangChain, advanced vector databases, Experience with Azure OpenAI, Google Vertex AI, or AWS Bedrock

Responsibilities

• Develop and implement AI and machine learning strategies to drive healthcare solutions, build and test machine learning, deep learning, and Generative AI models, and RAG pipelines
• Research, design and implement scalable machine learning models including efficient programming and multi-threading
• Engage in the entire software development lifecycle, from design and testing to deployment and maintenance
• Build robust machine learning pipelines to support production processes
• Process and transform source data to be used in machine learning pipelines, leveraging cloud computing
• Driving adoption of new science and best practices among service teams by design and implementation of advanced state-of-the art solutions utilizing Large Language Models (LLMs)
• Collaborate with cross-functional teams to determine applicability of AI to business problems
• Demonstrate self-driven initiative by taking ownership and creating end-to-end solutions
• Develop and maintain code libraries, tools, and frameworks to support generative AI development

Benefits

• 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.

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level