tensorviz.tensorviz-vscode

tensorViz — PyTorch & JAX Model Visualizer

Total downloads

366

7d gain

232

Rolling weekly delta

Rating

- / 5

0 reviews

Version

0.1.13

Latest published

Activity

Extension trends

Manifest

Metadata

Namespace: tensorviz
Name: tensorviz-vscode
License: MIT
Homepage: https://tensorviz.dev/
Repository: https://github.com/tensorviz-org/tensorviz-examples.git
Reviews: 0

Taxonomy

Categories

Visualization Machine Learning

Description

Explore PyTorch and JAX/Flax model graphs, inspect tensor shapes, jump to Python source, and review PyTorch architecture changes in VS Code.

Archive

Release history

2 latest releases