AI/ML Platform Engineer
Wiraa · United States · Today
RemoteRemoteInformation TechnologyFull-time
About The Company Optum Tech is a global leader in health care innovation dedicated to developing advanced solutions that enhance individual well-being and improve the efficiency of the healthcare system. Our teams leverage cutting-edge technologies such as data analytics, artificial intelligence (AI), machine learning (ML), and cybersecurity to address some of healthcare's most complex challenges. We are committed to making a meaningful impact on lives by fostering innovation, collaboration, and growth. At Optum Tech, your contributions have the potential to drive transformative change in healthcare delivery and patient outcomes. Join us to be part of a forward-thinking organization that values innovation, diversity, and a shared mission to improve health worldwide. About The Role As a Principal AI/ML Platform Engineer at Optum Tech within the UnitedHealth Group enterprise team, you will serve as a key architect and leader in the development and management of our AI compute platforms. Your role involves owning the technical direction, reference architecture, and strategic evolution of our multi-tenant AI infrastructure, which spans both on-premises and cloud environments. You will be responsible for designing scalable, secure, and compliant AI platforms that support high-performance model training and inference workflows. This includes managing advanced compute estates with NVIDIA GPUs, OpenShift clusters, and public cloud AI services such as Azure AI Foundry, AWS Bedrock, and GCP Vertex AI. Your expertise will help optimize resource utilization, enforce security standards, and ensure compliance with healthcare regulations including HIPAA. This position offers the flexibility to work remotely from anywhere within the U.S., with specific in-office requirements for candidates located in Minneapolis or Washington, D.C. Qualifications Bachelor’s degree in Computer Science, Engineering, or related field, or 4+ years of equivalent experience10+ years of experience in infrastructure, DevOps, SRE, or ML platform engineering5+ years of experience operating Kubernetes or OpenShift at scale in production environments3+ years of experience designing and managing accelerated-compute (AI/GPU) infrastructure with NVIDIA GPU Operator, MIG/time-slicing, and DCGM3+ years of experience architecting hybrid AI platforms spanning on-premises and cloud services (Azure, AWS, GCP)3+ years of experience with HPC/AI networking technologies such as InfiniBand, RoCEv2, GPUDirect RDMA, and NCCL3+ years of experience managing multi-cluster environments using RHACM and GitOps tools like Argo CD or Flux3+ years of experience implementing enterprise security controls including RBAC, OIDC/OAuth, Vault, and mTLS Responsibilities Define and own the reference architecture for multi-tenant AI compute platforms across hybrid environments, ensuring scalability, security, and complianceEstablish network, latency, and topology standards for distributed training workloads, optimizing interconnects such as NVLink, InfiniBand, and RoCEv2Implement cluster governance, automated lifecycle management, and GitOps workflows to ensure consistent and reliable operations of bare-metal and cloud infrastructureDesign and refine cost models, including accelerator-sharing strategies and cost-per-training metrics, to inform procurement and resource allocation decisionsStandardize model-serving and inference platforms, setting policies for quantization, provenance, and model routing to improve efficiency and traceabilityDevelop workload placement strategies balancing self-hosted versus managed cloud deployment based on data residency, latency, and compliance needsEnforce identity and access management controls, security policies, and secret management practices to safeguard AI workloads and autonomous agentsEstablish enterprise SLAs, disaster recovery plans, and upgrade strategies for AI infrastructure components such as OpenShift, GPU drivers, and firmwareCollaborate with cross-functional teams, including AI, privacy, and security stakeholders, to ensure seamless integration and HIPAA compliance across platforms Benefits Comprehensive benefits package including health, dental, and vision insuranceIncentive and recognition programs to reward outstanding performanceEquity stock purchase options and 401(k) contributionsFlexible remote work arrangements within the U.S.Opportunities for professional development and certificationsSupportive and innovative work environment committed to diversity and inclusion Equal Opportunity UnitedHealth Group is an Equal Employment Opportunity employer. We value diversity and are committed to creating an inclusive environment for all employees. Qualified applicants will receive consideration without regard to race, national origin, religion, age, sex, sexual orientation, gender identity, disability, veteran status, or any other protected characteristic under applicable law. We adhere to all relevant laws and regulations and promote a workplace free from discrimination and harassment. Our organization is dedicated to providing equal opportunities for growth and success to all team members.