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

• 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

• 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