I found:

  
      
      GitHub
  
  
    

moskomule/anatome 24

Ἀνατομή is a PyTorch library to analyze representation of neural networks - moskomule/anatome


  
  
    
    
  
  


but it doesn’t seem to work reliably (or is it documented).
Is there a standard library pytorch peeps use for cca, cka, neural net comparison analysis that actually works? (ideally on gpu)