Senior AI/ML Engineer

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.

Primary Responsibilities:

• Model Development & Deployment
  • Design, develop, and deploy machine learning models and algorithms into production environments
  • Apply deep learning technologies for tasks such as computer vision, NLP, NLU, ASR, and semantic understanding
• AI Frameworks & Tools
  • Work with agentic AI frameworks like LangChain and LangGraph
  • Optimize and quantize GenAI models, including those using RAG, Transformers, and VectorDBs
• Data Engineering & Analysis
  • Handle large-scale training and inference using platforms like Databricks and PySpark
  • Perform statistical analysis, feature engineering, and dimensionality reduction (e.g., PCA, SVD, UMAP, t-SNE)
• Software Engineering Integration
  • Build scalable AI platforms and integrate ML solutions into existing systems using REST APIs, Docker, and cloud services (Azure, AWS)
  • Collaborate with cross-functional teams to align AI architecture with business goals
• AI Governance & Strategy
  • Participate in AI governance processes including intake, proof-of-concept, and capital planning
  • Break down large initiatives into manageable, cost-effective projects with clear KPIs
• Mentorship & Collaboration
  • Mentor junior AI/ML engineers and foster a collaborative, feedback-driven team culture
  • Engage in code reviews, pair programming, and community-based education initiatives
• 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 regard 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:

• Undergraduate degree or equivalent experience
• 8+ years of total experience
• 3+ years of work experience in AI projects with deployment of AI based projects in production
• Experience implementing multiple machine learning and AI techniques such as supervised, unsupervised, reinforcement learning, Deep learning, NLP
• Proven experience in containerizing AI/ML projects
• Proven experience in CI/CD pipelines and tools like GitHub Actions
• Good Understanding of Agentic AI, and Agentic Frameworks (LangGraph, LangChain)
• Knowledge and Experience of working with Large Language Models, RAG, Transformers etc
• Develop sophisticated ML models to create new data samples, patterns, or content based on existing data or inputs
• Should have a working experience of AI/ML projects deployed in Production
• Proficiency with cloud development environments such as Azure Proficiency with experience in ML and NLP
• Comprehensive knowledge and experience of Azure/AWS services for AI/ML Projects (Databricks, AKS/EKS, ADF, Cognitive Services etc, Logic Apps, Power Automate etc.)
• Proven good grasp of using LLMs, Deploying LLMs Proficiency in libraries such as Hugging face and OpenAI API
• Conduct model testing, validation, and optimization to enhance performance
• Collaborate with stakeholders to preprocess, analyze, and interpret extensive datasets
• Stay current with the latest advancements in generative AI technologies and practices
• Orchestration and management of containers in production environments for AI/ML projects
• Proven strategic thinking in systems capacity planning, hardware and software updates, and migration
• Solid programming skills in Python
• Familiarity with AIML governance, ethics, and responsible AI practices
• Proven track record in resolving code vulnerabilities

Preferred Qualifications:

• Certifications in AI/ML
• Experience in high level cloud architecture design for AI/ML projects
• Knowledge of Git Management 
• Familiarity with cloud platforms (e.g., AWS, Google Cloud, Azure)

• Knowledge of UAIS and MLRB 

   

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.