Augmented ML Workflow