Augmented AI frameworks