
6/26/2024 · Kristen Wright, Eric Yang
What this post added
Introduced Model Explorer, a graph visualization tool for ML models, now publicly available as part of the Google AI Edge family. Model Explorer supports large model graphs (tens of thousands of nodes) with 60 FPS visualization, integrates with JAX, PyTorch, TensorFlow, and TensorFlow Lite, runs in Colab, and is extensible. It offers node metadata overlays for debugging performance and numeric accuracy, and side-by-side graph comparison for conversion error debugging. It is used by Waymo and Google Silicon for optimizing models like Gemini Nano.