willyfh/visualtorch

VisualTorch aims to help visualize Torch-based neural network architectures.

What it solves

VisualTorch は、PyTorch ニューラルネットワークアーキテクチャを可視化する方法を提供し、開発者や研究者がモデル構造を理解、ドキュメント化、および提示するための作業を容易にします。単純なシーケンシャルモデルから、より複雑なカスタムモデルまでサポートしています。

How it works

このツールは、トレーシングベース（tracing-based）のアプローチを使用し、ダミー入力を用いて実際のフォワードパスを実行することで、モデルのアーキテクチャをマッピングします。その後、これらをフロー形式、グラフ形式、LeNet形式の 3 つの異なる視覚的スタイルでレンダリングします。さらに、モデルがレイヤーごとに構築される様子を示すアニメーション GIF を生成できます。

Who it’s for

研究者や、論文、プレゼンテーション、または社内ドキュメント用に、ニューラルネットワークのプロフェッショナルな図解を作成する必要がある PyTorch ユーザー（研究者や開発者を含む）向けに設計されています。

Highlights

  • Multiple Visualization Styles: フロー、グラフ、および LeNet スタイルの図解をサポートしています。
  • Animated Reveals: モデルアーキテクチャをアニメーション GIF としてレンダリングする機能があります。
  • MCP Integration: stdio を介してモデルのソースコードから図解を生成するための、オプションの MCP server を含んでいます。
  • Research-Backed: Nature, IEEE, および MDPI に掲載された研究で使用されています。