10.1101/2023.03.16.532916

TorchLens: A Python package for extracting and visualizing hidden activations of PyTorch models