Lead AI / ML Engineer - Remote

Optum is a global organization that delivers care, aided by technology to help millions of people live healthier lives. This role is a senior, hands-on AI/ML engineering position focused on building, enhancing, and integrating Machine Learning and Generative AI solutions within Databricks.

Responsibilities:

• Designing and implementing ML models
• Creating custom embeddings
• Building RAG pipelines
• Implementing vector databases
• Developing agentic workflows with LangChain
• Integrating AI into enterprise applications for Claims Payment Integrity (PI)
• The role partners closely with Data Scientists, Product Teams, and Platform Teams to deploy AI-driven intelligence across PI use cases
• Contribute to experimentation with novel ML and Generative AI architectures
• Create and maintain embedding models (domain-specific embeddings, document/text embeddings)
• Support Retrieval‑Augmented Generation (RAG) solutions, including semantic search, vector databases, and LLM integration
• Help design and implement AI‑powered enterprise applications
• Analyze large datasets to identify trends, patterns, and actionable insights
• Partner with AI/ML Engineers and Leads to enhance existing models and improve model accuracy and performance
• Optimize AI pipelines for scalability, performance, reliability, and cost efficiency
• Integrate AI/ML solutions into enterprise systems by partnering closely with data engineering, product, and platform teams
• Collaborate effectively with cross-functional teams and stakeholders, communicating complex technical concepts clearly
• Contribute to skill development for engineers and interns in AI/LLM techniques
• Comply with company policies, procedures, and directives