Torchvista: Building an Interactive Pytorch Visualization Package for Notebooks | Towards Data Science

Building a tool to interactively visualize the forward pass of any Pytorch model from within notebooks.