Explore PyTorch model graphs, tensor shapes and source code in VS Code.
About
We build tensorViz, a free VS Code extension for exploring PyTorch model graphs, captured tensor shapes, source code and architecture changes. Here we share practical model-inspection walkthroughs. Try browser examples at https://tensorviz.dev/explore/.