torchvista

torchvista is an open-source tool that enables interactive visualization of PyTorch forward passes in Jupyter notebooks, maintained as a public GitHub repository under the sachinhosmani organization. It supports interactive drag-and-zoom graphs, collapsible nodes for hierarchical modules, and error-tolerant partial visualization when errors arise.

GitHub repo Directory import
0 shouts Added Sep 23, 2026 github.com
github.com/sachinhosmani/torchvista
torchvista website preview

Links

Tags