cka
Compare neural networks by their feature similarity
Centered Kernel Alignment (CKA) with Efficient Computation and Layer-wise Visualization for PyTorch
[NeurIPS'22] PyTorch library to compare similarity between NN representations
