that looks promising! Do they have CCA and CKA? I couldn’t find them in their library 

reddit crosspost: https://www.reddit.com/r/pytorch/comments/ksocf8/what_is_a_good_cca_cka_library_for_pytorch_that/ 24

link to captum github page: GitHub - pytorch/captum: Model interpretability and understanding for PyTorch 20