Torchvista: Building an Interactive Pytorch Visualization Package for Notebooks

Torchvista introduces an interactive visualization package designed for Jupyter notebooks that enables users to dynamically explore the forward pass of any PyTorch model. This tool enhances model interpretability by providing real-time, visual insights into the data flow through neural network layers, facilitating debugging and understanding of complex architectures within an accessible, notebook-based environment.