torch-cka

A package to compare neural networks by their feature similarity