ENH: Add per-stage initialization possiblity for Similarity transform