AI Edge Model Explorer

0.1.32 verified Fri May 01 auth: no python

A modern model graph visualizer and debugger for TensorFlow Lite, MediaPipe, and other ML models. Provides an interactive graph viewer to inspect model structure, tensors, and operations. Current version 0.1.32, pre-release state with frequent updates.

pip install ai-edge-model-explorer

Common errors

error ModuleNotFoundError: No module named 'ai_edge_model_explorer' ↓

cause Incorrect import path; the actual package is under 'google.ai.model_explorer'.

fix

Use 'from google.ai import model_explorer'

error AttributeError: module 'google.ai.model_explorer' has no attribute 'visualize' ↓

cause Older version (pre-0.1.15) used 'view' instead of 'visualize'.

fix

Update to >=0.1.15 or use 'model_explorer.view()'.

Warnings

breaking Import path changed after 0.1.0: use 'from google.ai import model_explorer' instead of 'import ai_edge_model_explorer'. ↓

fix Update imports to use the google.ai namespace.

gotcha The library requires a modern browser. It may not work in headless environments without a display server (e.g., remote SSH). ↓

fix Use a virtual display (like Xvfb) or ensure a desktop environment.

deprecated Support for TensorFlow 1.x models is deprecated in 0.1.20+. Models must be exported as TF2 SavedModel or TFLite. ↓

fix Convert models to TF2 format or use TFLiteConverter.

Imports

explorer
wrong
```
import ai_edge_model_explorer
```
correct
```
from google.ai import model_explorer
```
Top-level package is 'google.ai.model_explorer', not the PyPI name.

ModelExplorer

wrong

from ai_edge_model_explorer import ModelExplorer

correct

from google.ai.model_explorer import ModelExplorer

Class is within the google namespace.

Quickstart

Load a TFLite model and open the graph visualizer in a web browser.

from google.ai import model_explorer

# Launch the interactive viewer (opens browser)
model_explorer.visualize('path/to/model.tflite')