torchvista

github.com ↗

Interactive notebook-based tool to visualize the forward pass of any PyTorch model.