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…