sachinhosmani/torchvista
An interactive visualization tool for inspecting PyTorch model architectures and forward passes directly in Jupyter and Colab notebooks.
Collecting fresh signals — velocity needs a few days of history.
collecting data…
star history
torchvista provides a single-line API to generate interactive visualizations of PyTorch model computation graphs within web-based notebooks. It supports collapsing hierarchical modules, error-tolerant partial visualization when shape mismatches occur, and allows users to inspect node-level parameter and attribute information. The tool exports visualizations as images, SVG, or HTML for documentation and sharing.