model-explorer
Model Explorer是一款AI模型结构可视化工具,支持TFLite、TF、TFJS、MLIR和PyTorch等多种格式。它提供层级化展示,允许动态展开或折叠各层,并具备高亮输入输出、元数据叠加、交互式弹窗等功能。工具还支持搜索、显示相同层和GPU加速渲染,方便开发者进行模型探索和调试。Model Explorer提供扩展框架,便于添加对其他格式的支持。