Extracting and visualizing hidden activations and computational graphs of PyTorch models with TorchLens
Scientific Reports - Extracting and visualizing hidden activations and computational graphs of PyTorch models with TorchLens
https://www.nature.com/articles/s41598-023-40807-0