AI for workflow optimization