tensorViz — PyTorch Model Visualizer

Explore PyTorch models as interactive graphs with the free tensorViz VS Code extension. Expand nested modules, inspect shapes captured from a forward pass, jump to Python source and review architecture changes beside code diffs. The website includes recorded model graphs to explore without installation. Local capture requires Python, model dependencies and example inputs.