Centered Kernel Alignment (CKA) with Efficient Computation and Layer-wise Visualization for PyTorch
pip install pytorch-cka