GitHub repository: vishalsingh17/surgeon-pytorch
Description: A library to inspect and extract intermediate layers of PyTorch models.