Associate AI/ML Engineer

Requisition Number: 2388586
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:

• Help design and develop the next generation of NLP, ML & AI products, and services for healthcare
• Develop machine learning and deep learning models and systems in domains including, but not limited to: NLP, NLU, NLG, SLU and multidimensional time series forecasting among others
• Exposure in RAG, LangChain, VectorDBs
• Ability to Quantize, Optimize GenAI models
• Manage NLP & ML models lifecycle for a suite of products
• Run large complex proof-of-concepts for the healthcare business
• Manage prioritization and technology work for building NLP, ML & AI solutions
• Lead the full end-to-end machine learning development process including data ingestion and preparation, feature engineering, analysis and modeling, model deployment, performance tracking and documentation
• Establish best practices for end-to-end deep learning and machine learning development cycle to ensure rigor in process and quality in outcome
• Work with a great deal of autonomy to find solutions to complex problems
• Design and architect scalable ML solutions for complex business problems
• Lead Projects: Lead ML projects from conception to deployment, ensuring timely delivery and quality
• Mentorship: Mentor junior engineers and provide technical guidance to the team
• Research and Innovation: Stay updated with the latest research in AI/ML and implement cutting-edge techniques.
• Cross-functional Collaboration: Collaborate with cross-functional teams to align ML solutions with business goals
• Comply with all applicable Company policies, procedures, and business directives, changes including those relating to work location, team assignments, work schedules, and flexible work arrangements

Required Qualifications:

• Bachelor's or Master's degree in Computer Science, Data Science, Statistics, or a related field
• 3+ years of experience in AI/ML
• Solid programming skills in Python, with experience in pandas, numpy, and SQL
• Experience with the full life cycle of AI/ML projects, including EDA, model development, tuning, and drift analysis
• Hands-on experience in delivering production-grade AI/ML projects
  Exposure to RAG modeling using vector databases (e.g., Milvus, Chroma, FAISS)
• Solid knowledge of discriminative AI model development and generative AI applications
• Familiarity with Python frameworks such as Flask API and FastAPI
  Math behind ML and DL algorithms
• Proficiency in frameworks such as Scikit Learn, TensorFlow, and PyTorch

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.