tensorviz

Explore PyTorch models in VS Code with tensorViz. Follow tensor shapes from captured runs, inspect nested layers, jump to source, and review model changes.

Watch practical walkthroughs, then try the free extension or browser examples at tensorviz.dev.