inspired from this paper as a potentiall better similarity representation metric: [2108.01661] Grounding Representation Similarity with Statistical Testing 1

note scipy does have it: scipy.linalg.orthogonal_procrustes — SciPy v1.7.1 Manual 1