Optum is a global organization that delivers care, aided by technology to help millions of people live healthier lives. The work you do with our team will directly improve health outcomes by connecting people with the care, pharmacy benefits, data and resources they need to feel their best. Here, you will find a culture guided by inclusion, talented peers, comprehensive benefits and career development opportunities. Come make an impact on the communities we serve as you help us advance health optimization on a global scale. Join us to start Caring. Connecting. Growing together.

We are seeking a seasoned AI/ML Architect to lead design, development and deployment of advanced machine-learning solutions. In this role you will define end-to-end ML architecture, select appropriate tools and frameworks, drive POCs and guide engineering teams in productionizing scalable AI services. A solid foundation in statistics, deep learning and generative AI, hands-on cloud expertise, and exceptional communication skills are essential.

Primary Responsibilities:

Architect, design and implement end-to-end ML/DL workflows: data ingestion, feature engineering, model training, evaluation, deployment and monitoring Lead proof-of-concept experiments in generative AI (transformers, GANs, diffusion models) to solve business problems Define cloud-native ML infrastructure on Azure, AWS or GCP: containerization (Docker/Kubernetes), ML pipelines (SageMaker, Vertex AI, Azure ML), MLOps (CI/CD, model registry, monitoring) Establish best practices for model governance, versioning, reproducibility and security Collaborate with data engineers, data scientists, software engineers and product managers to translate business requirements into technical solutions Mentor and coach mid- to junior-level engineers and foster a culture of continuous learning Evaluate emerging tools, libraries and research to drive innovation and maintain competitive edge Document architecture designs, conduct design reviews and present technical proposals to stakeholders Comply with the terms and conditions of the employment contract, company policies and procedures, and any and all directives (such as, but not limited to, transfer and/or re-assignment to different work locations, change in teams and/or work shifts, policies in regards to flexibility of work benefits and/or work environment, alternative work arrangements, and other decisions that may arise due to the changing business environment). The Company may adopt, vary or rescind these policies and directives in its absolute discretion and without any limitation (implied or otherwise) on its ability to do so

Required Qualifications:

Bachelor’s, Master’s or PhD in Computer Science, Statistics, Applied Mathematics, Electrical Engineering or related field 10+ years of professional experience in machine learning, data science or AI roles Deep expertise in core ML and statistical methods: supervised/unsupervised learning, regression, classification, clustering, time series, Bayesian modeling Hands-on experience with Python and libraries: pandas, NumPy, scikit-learn Proficiency in deep learning frameworks: TensorFlow, Keras, PyTorch Practical experience building or fine-tuning generative models (e.g., GPT, BERT, Stable Diffusion, custom architectures) Solid background in probability, linear algebra and statistical inference Demonstrated experience with cloud ML services and infrastructure design on at least one major cloud platform (AWS, Azure or GCP) Track record of moving models from research/POC into production at scale Proven excellent problem-solving ability and solid verbal/written communication skills

Preferred Qualifications:

Experience in the Healthcare domain: familiarity with clinical data standards (HL7, FHIR), regulatory requirements (HIPAA) and EHR systems Experience with MLOps tools: MLflow, Kubeflow, TFX, Airflow or equivalent Knowledge of data visualization tools (Tableau, Power BI) and dashboarding Familiarity with big data technologies: Apache Spark, Hadoop, Dask Contributions to open-source AI/ML projects or publications in peer-reviewed venues

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.