Skip to content

google-ai-edge/model-explorer

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Model Explorer PyPI version

Model Explorer offers an intuitive and hierarchical visualization of model graphs. It organizes model operations into nested layers, enabling users to dynamically expand or collapse these layers. It also provides a range of features to facilitate model exploration and debugging, including the ability to highlight input and output operations, overlay metadata on nodes, display layers in interactive pop-ups, perform searches, show identical layers, GPU-accelerated graph rendering, among others. It currently supports TFLite, TF, TFJS, MLIR, and PyTorch (Exported Program) model format, and provides an extension framework for developers to easily add support for additional formats.

Home page screenshot


To start using Model Explorer, run:

$ pip install ai-edge-model-explorer
$ model-explorer

Or give it a try on Hugging Face (only support visualizing uploaded models)

Please check out our Wiki for more details:

We invite you to participate in research studies on Model Explorer. Sign up here.

Community-contributed adapters

Contributions

You are invited to create custom adapters to add support for additional model formats. Please refer to our Develop Adapter Extension guide. We will gladly consider pull requests to add links to your adapter's GitHub repository and PyPI package to this README.

Resources