This is a fully remote, mid-level position with UnitedHealth Group's Optum AI unit. The role involves building machine learning infrastructure for healthcare, focusing on system design and scalable innovation.
Skills / Requirements
- AI Governance
- Airflow
- AWS SageMaker
- Azure ML
- Docker
- GCP Vertex AI
- Kubernetes
- LangChain
- LLM Stacks
- MLOps
- Prompt Chaining
- Python
- PyTorch
- Spark
- TensorFlow
- Vibe Coding
Why Apply
This role is ideal for developers who use AI tools like Cursor and Claude Code for code generation. You'll manage LLM stacks with LangChain and LlamaIndex, focusing on prompt chaining and architecture.
What You'll Be Doing
You'll design and build MLOps pipelines and LLM gateways, ensuring secure and scalable machine learning solutions in healthcare. Your focus will be on high-level system design rather than manual coding tasks.
Working in Eden Prairie, Minnesota
- Eden Prairie offers a suburban lifestyle with access to Minneapolis's tech scene.
- The area is known for its high quality of life and strong school systems.
- Remote work provides flexibility to collaborate asynchronously across time zones.
- Fully remote roles allow you to work from anywhere, offering a better work-life balance.
Pay and Career Growth
The salary range of $98,547 to $175,978 indicates competitive compensation. UnitedHealth Group offers a culture that encourages innovation through AI-assisted tools, providing a clear path for career advancement in healthcare AI.
Benefits and Perks
- AI-assisted tools
- cloud-native environment
- Remote Work
- vibe coding
Is This Role Right for You?
Good fit if you...
- You thrive in cloud-native environments and are comfortable with AWS, Azure, or GCP.
- You enjoy using AI tools to automate code generation and focus on architecture.
- You have experience with Python, PyTorch, and TensorFlow in a professional setting.
May not be for you if...
- You prefer on-site work and face-to-face collaboration.
- You are uncomfortable with AI-assisted development tools.
- You lack experience in building scalable machine learning infrastructure.
Original Job Description
Architecting the Future of Healthcare AI
In this role, you’ll join Optum’s elite AI unit to construct the foundation for machine learning across a massive healthcare ecosystem. You won’t just be training models; you’ll be building the sophisticated infrastructure—from MLOps pipelines to LLM gateways—that enables secure and scalable healthcare innovation. This is a chance to move beyond manual boilerplate and focus on high-level system design.
The Tech Stack
You will work with a modern, cloud-native environment designed for rapid iteration:
Core Engineering: Python, PyTorch, and TensorFlow deployed via Docker and Kubernetes.
AI Orchestration: Managing LLM stacks using LangChain and LlamaIndex to optimize performance and costs.
Cloud Platforms: Deep integration with AWS SageMaker, Azure ML, or GCP Vertex AI.
Data Pipelines: Leveraging Spark and Airflow to handle distributed data at an enterprise scale.
A Vibe-First Engineering Culture
What sets this position apart is the explicit encouragement to use AI-assisted development tools like Cursor and Claude Code. UnitedHealth Group recognizes that the fastest way to solve complex healthcare problems is through ‘vibe coding’—leveraging LLMs to handle the heavy lifting of code generation while you focus on prompt chaining, architecture, and responsible AI governance.