VisualTorch — MCP-сервер для Claude Code: диаграммы и установка
VisualTorch строит диаграммы архитектуры PyTorch и анимированные GIF по исходному коду модели. Пригодится для разбора структуры нейросети.
claude mcp add visualtorch -- uvx visualtorchЧто делает VisualTorch
VisualTorch — MCP-сервер для визуализации моделей PyTorch. Он строит архитектуры в стилях graph, flow и lenet, а также создаёт анимированные GIF, где модель показывается по слоям или колонкам. Сервер работает через stdio и передаёт метаданные структурированного результата.
Для работы нужны Python 3.10+ и PyTorch 2.0+. Установка MCP-зависимостей описана через пакет visualtorch[mcp], а для подключения к Claude Code используется команда с uvx. Сервер умеет работать с моделями Sequential и пользовательскими моделями. Он запускает реальный forward pass, поэтому при ветвлении по значениям тензора в диаграмму попадёт только ветка, которую выбрал тестовый вход. Для нескольких выходных тензоров размер узла считается по первому, хотя формы всех выходов можно показать через show_dimension=True. Это ограничение стоит учитывать при разборе моделей с условной логикой.
Какие задачи решает
- построить диаграмму архитектуры PyTorch-модели
- показать связи между слоями в стиле
graph - сгенерировать схему в стилях
flowилиlenet - создать GIF с поэтапным раскрытием модели
- проверить структуру пользовательской модели через forward pass
Примеры запросов агенту
- Построй диаграмму модели из
model.pyв стилеgraphи покажи размеры тензоров. - Сгенерируй для этой PyTorch-модели анимированный GIF в стиле
flow. - Сравни визуальное представление модели в стилях
graph,flowиlenet.
Как установить VisualTorch
Командой платформы
claude mcp add visualtorch -- uvx visualtorchuvx запускает команду с именем пакета; если в README она другая — uvx --from visualtorch <команда>
Командой платформы
codex mcp add visualtorch -- uvx visualtorchuvx запускает команду с именем пакета; если в README она другая — uvx --from visualtorch <команда>
Файл настроек
{
"mcpServers": {
"visualtorch": {
"command": "uvx",
"args": [
"visualtorch"
]
}
}
}uvx запускает команду с именем пакета; если в README она другая — uvx --from visualtorch <команда>
Файл настроек
{
"servers": {
"visualtorch": {
"type": "stdio",
"command": "uvx",
"args": [
"visualtorch"
]
}
}
}uvx запускает команду с именем пакета; если в README она другая — uvx --from visualtorch <команда>
Командой платформы
gemini mcp add visualtorch uvx visualtorchuvx запускает команду с именем пакета; если в README она другая — uvx --from visualtorch <команда>
По шагам, с командами для каждой платформы: Как подключить MCP-сервер: Claude Code, Codex, Cursor, VS Code, Gemini CLI и другие
Похожие
Рядом в теме «ИИ и ML»
Все MCP-серверы по теме «ИИ и ML» →Вопросы о VisualTorch
Как установить VisualTorch в Claude Code?
Выполните claude mcp add visualtorch -- uvx visualtorch.
Что делает VisualTorch?
VisualTorch строит диаграммы архитектуры PyTorch в стилях graph, flow и lenet, а также создаёт анимированные GIF с поэтапным показом модели.
Какие ограничения есть у VisualTorch при трассировке модели?
Сервер анализирует реальный forward pass. При ветвлении, зависящем от значения тензора, отображается ветка, выбранная тестовым входом; для нескольких выходов размер узла берётся по первому тензору.
