Torchvista

github.com ↗

- Torchvista is an interactive notebook-based tool to visualise the forward pass of any PyTorch model as a computation graph within notebooks, with support for collapsible nested modules and error-tolerant partial visualisations.