VisualTorch MCP server
Render PyTorch architecture diagrams and animated GIF reveals from trusted model source.
332 stars227 downloads/wk
Reviews
Write oneNobody has reviewed VisualTorch yet.
If you have run it, two minutes of your experience saves the next person an afternoon.
VisualTorch tools (4)
write = sends, deletes, buys or postsRead from the package source without running it. The installed server may list more.
animate_modelRender a trusted PyTorch model source as an animated VisualTorch GIF.
visualize_modelRender a trusted PyTorch model source as a static VisualTorch PNG.
visualtorch_capabilitiesReturn structured styles, options, palettes, output contracts, and safety constraints.
visualtorch_referenceReturn canonical upstream VisualTorch documentation links for all or one style.
Public scan report
scanner v0.1.9 · 2026-09-24 · same rubric, same numbers if you re-run it
- Code scan24 source files scanned20/25
- –Live reliabilityno gateway calls yet and no remote to proben/a
- –Tool poisoningtools not inspected (local package is not executed); not countedn/a
- Auth qualitylocal package, no credentials required12/15
- Maintenancelast push 1 days ago15/15
- Maintainer identityregistry namespace matches repository owner; GitHub account older than a year8/10
Findings (1)
- mediumeval / new Function used
exec.evalvisualtorch-1.4.1/visualtorch_mcp/worker.py: … raise TypeError(message) model = eval(model_expression, namespace) # noqa: S3…
What the publisher says
From the VisualTorch repository's README, as published. We do not edit it. Read it on GitHub
VisualTorch aims to help visualize Torch-based neural network architectures. It currently supports generating flow-style, graph-style, and LeNet-style architectures for PyTorch Sequential and Custom models. Its original visual styles were inspired by visualkeras, pytorchviz, pytorch-summary, and torchview; since then, it has grown its own unified tracing backend and architecture-handling logic well beyond its origins.
Note: 1.0+ is a major release with breaking API changes, but with significantly better features and algorithms - upgrading is recommended. For the old API, use 0.2.5 or older.
Limitation: VisualTorch traces a real forward pass to build the diagram, which has an inherent limitation shared by any tracing-based approach (not a bug, and not fixable without full symbolic execution): models with data-dependent control flow (e.g. a branch only taken if a tensor value crosses some threshold) only show whichever branch the traced dummy input happened to take. Separately, a layer that returns multiple meaningful output tensors (e.g. a custom multi-task head, or nn.LSTM's (output, (hn, cn))) still has its node's size based on only its first tensor; with show_dimension=True, every output tensor's shape is shown in the label, not just the first. Downstream connections are correct either way. Contributions are welcome!
Animated Reveal
Every style can also render as an animated GIF, revealing the model one layer/column at a time, via visualtorch.animate(model, input_shape, style=...) - see it in action for flow, graph, and lenet styles.
Documentation
Online documentation is available at visualtorch.readthedocs.io.
The docs include usage examples, API references, and other useful information.
Installation
See the Installation page.
MCP integration
Shortened. The full README is on GitHub.
Nothing above is checked by us. What we check is on the safety report.
Install directly
claude mcp add visualtorch -- uvx visualtorch
VisualTorch: common questions
- Is VisualTorch MCP server safe?
- Yes, by our scan: it is graded A (85/100). Read the VisualTorch safety report
- How do I install VisualTorch?
- It runs on your machine. Copy the Claude Code, Claude Desktop or Cursor config from the install section.
- Does VisualTorch need an API key?
- Not as far as the registry entry and our scan can tell: no credentials are declared or required.
- Is VisualTorch maintained?
- The last commit was in the last day (2026-09-24). The latest release is v1.4.1.
- What can I use instead of VisualTorch?
- Servers from other publishers that do the same job: Agent-Native Plan MCP server, three.ws Avatars MCP server and UI Beats MCP server. Compare all VisualTorch alternatives.
Alternatives to VisualTorch
Same job from other publishers: the closest match first, then the best rated.
- Agent-Native PlanStructured visual plans and PR recaps with diagrams, prototypes, annotations, and sharingnot reviewedEstablishedA
- three.ws Avatarsthree.ws 3D avatars in your agent: render a live, interactive 3D avatar inline, or get an embed.not reviewedEstablishedA
- UI BeatsMCP server for UI Beats. Let your coding agent search, read and install animated React components.not reviewedEstablishedA
- LeClapCompose and render video from one JSON template. Deterministic, runs locally, no generative model.not reviewedGrowingA