TensorViz documentation

Explore layers, follow tensor shapes, and find the matching Python. Open your model in your coding agent or editor.

Open your first graph →

Explore the guides

Bring the graph into your agent.

Install once, then ask your agent to open a model.

Plugin preview · Apple Silicon Mac · CPython 3.12 · CPU capture

Terminal

codex plugin marketplace add tensorviz-org/tensorviz-plugins
codex plugin add tensorviz@tensorviz-standalone

Run both commands in a terminal with Codex installed. Open your model project and start a new chat.

Plugin setup, permissions & other clients →

From your first graph to the details

Explore the graph

Expand modules, find a layer, and follow the connections. Inspect the shapes recorded by a capture.

Find your way around →

Follow the code

Select a layer to find its Python. Keep the architecture and its implementation in view.

Connect graph and source →

Review a change

Compare layers and connections between Python revisions, with the code diff beside the graph.

Read a graph diff →

Need a hand opening a model? Python environments · Model inputs · Installation options