#模型可视化
pytorch-summary
pytorch-summary提供类似Keras的model.summary()功能，帮助在PyTorch中可视化和调试模型。用户可以通过pip安装或从GitHub克隆获取，轻松查看模型参数和结构，支持多种输入格式。适用于各种神经网络模型，包括CNN和VGG16，支持计算模型大小和内存需求。该工具基于MIT许可，并由社区贡献者支持和启发。
VisualDL
VisualDL，一款集成于PaddlePaddle的高效可视化分析工具，能够展现训练参数、模型结构与数据样本的动态图表。完美支持Python，通过实时数据追踪、多实验视图对比等功能，助力开发者清晰直观地优化和提升模型表现。
penzai
Penzai是一个基于JAX的库，专为通过函数式pytree数据结构编写模型而设计，并提供丰富的工具用于可视化、修改和分析。适用于反向工程、模型组件剥离、内部激活检查、模型手术和调试等领域。Penzai包括Treescope交互式Python打印工具、JAX树和数组操作工具、声明式神经网络库及常见Transformer架构的模块化实现。该库简化了模型处理过程，为研究神经网络的内部机制与训练动态提供了支持。
model-explorer
Model Explorer是一款AI模型结构可视化工具，支持TFLite、TF、TFJS、MLIR和PyTorch等多种格式。它提供层级化展示，允许动态展开或折叠各层，并具备高亮输入输出、元数据叠加、交互式弹窗等功能。工具还支持搜索、显示相同层和GPU加速渲染，方便开发者进行模型探索和调试。Model Explorer提供扩展框架，便于添加对其他格式的支持。
NeuralFlow
NeuralFlow是一个Python工具，用于可视化Mistral 7B语言模型的中间层输出。它生成512x256的热图，展示模型每层的输出。该工具可用于分析模型结构和监控fine-tuning过程中的变化。NeuralFlow将4096维张量数据转化为直观的视觉表现，为AI模型开发提供新的分析方法。