UnitedHealth Group

Principal AI/ML Engineer

UnitedHealth Group$164K — $282K *
Healthcare
8 - 10 years of experience
Job Overview by Ladders

Qualifications

  • Bachelor's degree in computer science, engineering, or a technical field, or equivalent experience
  • 10+ years of professional software engineering experience building and operating production systems
  • 5+ years of experience designing distributed services and platform-level APIs
  • Experience building or operating ML, AI, or data-intensive systems in production
  • Hands-on experience with LLM-based or agentic systems
  • Expert-level proficiency in Python and another production language (Go, Java, etc.)
  • Experience with cloud-native delivery, containers, and Kubernetes

Responsibilities

  • Design, code, test, debug, document, and maintain AI platform services
  • Create platform public interfaces - APIs, SDKs, client libraries
  • Build capability gateways and implement Model Context Protocol servers
  • Develop and manage the agent harness and runtime
  • Establish engineering standards and CI/CD practices
  • Engineer platform for cost and scalability
  • Implement AI/ML capabilities in healthcare workflows

Benefits

  • Flexible remote work from anywhere in the U.S.
  • Opportunity for career growth and development
  • Comprehensive benefits package
  • Incentive and recognition programs
  • Access to equity stock purchase and 401k contributions
Full Job Description
As a principal engineer within the enterprise AI Platforms team within the Office of AI, you will design, build and operate the platform that healthcare agents run on. This is a software engineering role at its core applied to a domain where the workloads are models, retrieval and agents. You will own systems end to end: analysis and design, coding, code review, testing and debugging, documentation, release and delivery, and the long maintenance tail after launch.

Your work is enabling Optum clients with access to an AI Platform that provides secure, PHI-ready model access and favorable inference economics, a capability gateway that makes licensed intelligence callable from any approved agent or application, and the agent harness those agents run on. You will set the engineering standards, patterns and shared utilities other teams build on, and carry meaningful influence over the technical direction of the portfolio. Expect to prototype quickly with design partners, prove or disprove technical bets with evidence, and then harden what survives into platform-grade services that other engineers can easily adopt.

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.

Primary Responsibilities:

  • Design, code, test, debug, document and maintain the services that make up the AI platform and agent runtime, taking systems from design review through production operation and long-term support
  • Design the public surface of the platform - APIs, SDKs, client libraries, schemas and versioning strategy - so that internal teams, client engineering teams and external builders can integrate without bespoke work for each consumer
  • Build the Capability Gateway: implement Model Context Protocol (MCP) servers and tool interfaces with typed contracts, authentication and authorization, input and output validation, quota and rate limiting, audit logging, and an automated conformance test suite that new capabilities must pass
  • Build and operate the agent harness and runtime: orchestration and tool-calling execution, state and session management, retrieval paths, concurrency, queuing and retries, timeout and failure handling, idempotency, and graceful degradation under partial outage
  • Establish engineering standards, methods and tooling for the portfolio: coding standards, code review practice, branching and release strategy, CI/CD pipelines, infrastructure as code, environment management, and automated testing at unit, contract, integration and end-to-end levels
  • Engineer the platform for cost and scale, including inference routing across models, caching, batching, connection and resource pooling, capacity planning, and per-tenant cost telemetry that ties spend to completed work
  • Code and deploy the ML and LLM components of the platform into production: model and prompt serving, retrieval and embedding pipelines, adapters and fine-tuned variants, and the deployment mechanics behind them including canary, shadow and rollback paths
  • Implement AI/ML capabilities in support of healthcare workflows, which may include natural language processing and understanding, semantic search, intent classification, information extraction, document AI and computer vision, and automatic speech recognition applied to clinical notes, claims, faxes, referrals, prior authorizations and recorded encounters
  • Build the evaluation and experimentation infrastructure the platform depends on: golden datasets, task-level benchmarks, offline and online evals, human-in-the-loop review workflows, regression gates wired into CI, and experiment tooling with defensible statistical treatment of results
  • Work with large-scale computing frameworks and data analysis systems to build the data foundation behind agent capabilities, including distributed processing, feature and vector stores, streaming ingestion, schema evolution and lineage
  • Engineer for PHI from the first commit: data minimization and de-identification, tenant isolation, encryption in transit and at rest, secrets management, retention and access controls, and enforcement that keeps agent behavior inside approved data and action boundaries
  • Evaluate new tools, techniques and strategies - models, frameworks, orchestration approaches, serving stacks - with enough rigor to make a build, buy or adopt recommendation, and communicate the results and trade-offs to leadership and internal stakeholders
  • Partner across Platform Enablement, product management, design, security, privacy and legal, forward-deployed engineering, operations and external partners such as Azure and Databricks on one integrated backlog with explicit interfaces and SLOs
  • Raise the engineering bar through reference implementations, reusable libraries, design review participation, code review and technical mentorship, and influence thought and leadership on where the platform should go next


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:

  • Bachelor's degree in computer science, engineering or a related technical field, or equivalent experience
  • 10+ years of professional software engineering experience building and operating production systems
  • 5+ years of experience designing distributed services and platform-level APIs consumed by other engineering teams
  • Experience building or operating ML, AI or data-intensive systems in production, including responsibility for them after launch
  • Hands-on experience with LLM-based or agentic systems, including retrieval-augmented generation, tool and function calling, and orchestration
  • Expert-level proficiency in Python and at least one additional production language such as Go, Java, TypeScript, Scala or C++
  • Demonstrated depth in software engineering fundamentals: testing strategy, debugging complex production issues, performance profiling, code review and technical documentation
  • Experience with cloud-native delivery, including containers and Kubernetes, CI/CD, infrastructure as code, and observability tooling
  • Experience with large-scale computing frameworks and distributed data processing such as Spark/Databricks, Ray, Kafka or equivalent
  • Experience building software subject to security, privacy or regulatory constraints
  • Demonstrated ability to explain complex technical results and trade-offs clearly to non-technical stakeholders and senior leaders


Preferred Qualifications:

  • Experience building multi-tenant platforms or services, including tenant isolation, quota and entitlement enforcement, and developer experience as an explicit product concern
  • Experience implementing or operating Model Context Protocol (MCP) servers, tool-calling standards, or agent frameworks such as LangGraph or equivalent
  • Experience with model serving and optimization at scale, including vLLM, TensorRT/Triton, batching, KV-cache strategies, quantization and multi-model routing
  • Experience with vector databases, embedding lifecycle management, hybrid retrieval and enterprise knowledge graphs
  • Experience with inference cost management or FinOps for AI workloads
  • Experience with responsible AI practice, including bias and fairness evaluation, model risk management, model documentation, or alignment to frameworks such as the NIST AI Risk Management Framework
  • Experience on a 0-to-1 product or platform, working directly with design partners or customers to shape what gets built
  • Experience with Azure AI services and Databricks in a regulated enterprise environment
  • Experience with healthcare data and standards: claims, EHR data, FHIR, HL7, ICD, CPT, HCPCS, SNOMED CT, LOINC, and working with PHI inside HIPAA/HITRUST boundaries
  • Health care industry experience including provider, payer, medical device, pharmaceutical, or other health services a plus


*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 $164,600 to $282,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.

#optumtechpj

About UnitedHealth Group

UnitedHealth Group is a medical facility. They offer health technology, health finance, and pharmacy services. They are utilizing clinical data and intelligence to assist in the redesign, automation, and deployment of technology to streamline administrative operations and clinical decision-making. Payment systems for both consumers and providers are critical components of a health-care system.

UnitedHealth Group Careers

Joining UnitedHealth Group means becoming part of a diverse team dedicated to making a difference. As a leader in health services and innovation, our company offers a variety of job opportunities that allow professionals to leverage their skills, drive innovation, and improve lives. Work You’ll Do At UnitedHealth Group, you’ll contribute to a mission-focused environment where your expertise will influence the health and well-being of people worldwide. Our employment opportunities span across a wide range of disciplines, from healthcare specialists to data analysts, ensuring that your career journey is both dynamic and rewarding. Transform Healthcare with Your Expertise UnitedHealth Group stands at the forefront of health innovation. Our team collaborates to deliver solutions that lead to better patient outcomes. Working with us, you’ll find yourself at the intersection of technology, healthcare, and leadership, providing key insights that drive industry transformation. Join Our Global Team As part of our team, you’ll engage with over 300,000 professionals globally, dedicated to building a diverse and inclusive workplace. UnitedHealth Group is not just a company; it’s a community where you can grow your career through continuous learning and leadership opportunities. Our commitment to diversity training ensures that every team member can thrive. UnitedHealth Group Career Development We are committed to your professional growth. Explore career paths filled with promising job opportunities and internships that will harness your potential and expand your capabilities. Whether you’re a seasoned professional or a recent graduate, you’ll find that our career development programs support your ambition at every level. Innovative Work Environment At UnitedHealth Group, innovation is at the core of our operations. We encourage our team to bring forward-thinking ideas that challenge the status quo and lead to breakthrough improvements in patient care. Be Part of a Great Team Our culture fosters a collaborative and supportive environment where every member’s contribution is valued. Enjoy the benefits of being part of a global team that’s committed to making a difference in people’s lives. Future-Proof Your Career With UnitedHealth Group, your career is future-proof. Dive into a range of positions that offer both challenges and rewards. Our robust support system includes unmatched training, development programs, and certification support to propel your career forward. Stay Connected Join Our Team Discover the right position that matches your skills and interests. We are always hiring and look for passionate, curious, and solution-driven team players. Search UnitedHealth Group jobs today and take the first step towards a fulfilling career. Keep Up to Date Stay informed with career tips, insider perspectives, and industry-leading insights you can use today—all from the people who work here. Read Careers Blog Job Alert Emails Customize your subscription to receive job alerts, the latest news, and insider tips tailored to your preferences. Explore the exciting and rewarding opportunities that await at UnitedHealth Group.
Learn more about UnitedHealth Group
Size
350,000 employees
Market Cap
$493.1 billion
Industry
Net Income
$15.4 billion
Founded
1974
5 Year Trend
+9.2%
Revenue
$257.1 billion
NASDAQ

Similar Jobs

More Jobs at UnitedHealth Group

More Healthcare Jobs

Find similar Principal AI/ML Engineer jobs: