LEAD AI/ML ENGINEER - REMOTE

$104,700 - $190,400/Yr

Optum - Eden Prairie, MN

Full-time - Mid Level

Remote - Eden Prairie, MN

Insurance Carriers and Related Activities

About the position

The position involves working with Optum AI, a part of UnitedHealth Group, focusing on developing AI/ML solutions to improve health outcomes. The role requires building and testing machine learning models, developing robust pipelines, and collaborating with stakeholders to create impactful solutions. The work is centered around cutting-edge AI/ML research and development, with opportunities for innovation and publication.

Responsibilities

• Build and test deep learning and traditional ML models in distributed computing environments
• Develop robust machine learning pipelines and ML Ops procedures to support production processes
• Process and transform source data for machine learning pipelines using distributed computing and parallel processing
• Develop robust modules for evaluation and drift detection
• Design and plan experiments, conduct statistical analysis, and create data visualizations
• Communicate findings with business stakeholders and collaborate to develop solutions that meet customer needs
• Develop code and model documentation and assist with model governance approvals

Requirements

• Master's degree in computer science or related quantitative field
• 5+ years of industry/academic experience in machine learning or a related field
• 5+ years of AI/ML experience
• 5+ years of experience with Python, classical ML methods, and deep learning frameworks such as PyTorch or TensorFlow
• Experience with LLM frameworks and/or NLP approaches such as text embedding and text classification

Nice-to-haves

• Experience with distributed computing approaches such as Spark
• Experience with Snowflake and other relational databases
• Experience building Python libraries
• Experience working on collaborative software projects using GitHub
• Experience developing simple UIs for POCs demonstrating AI/ML models
• Proven solid technical communication skills

Benefits

• Comprehensive benefits package
• Incentive and recognition programs
• Equity stock purchase
• 401k contribution