VisualTorch for VS Code

io.github.willyfh/visualtorch

Render PyTorch architecture diagrams and animated GIF reveals from trusted model source.

client:VS Code transport:stdio runtime:pypi

Install VisualTorch in VS Code

.vscode/mcp.json

{
  "servers": {
    "visualtorch": {
      "type": "stdio",
      "command": "uvx",
      "args": [
        "visualtorch"
      ]
    }
  }
}

Requires agent mode. MCP: Open User Configuration puts the same block in every workspace. VS Code docs