AwesomeDataviz

Machine learning & AI visualization

Experiment tracking, model debugging, embeddings and LLM traces. 17 tools, ranked by GitHub stars and updated .

Tools for understanding machine learning systems: experiment tracking and training curves, model debugging, embedding projections and traces of LLM applications.

By GitHub stars, the most popular ML visualization tools here are Netron (33.5K stars), PlotNeuralNet (25K stars), and Opik (22.4K stars). 10 of 17 had commits in the last 90 days.
1 Netron

Viewer for neural network, deep learning and machine learning models.

 Machine learning 33.5K 926 MIT
2 PlotNeuralNet Inactive

LaTeX code for drawing neural network architecture diagrams.

 Machine learning 25K 0 MIT
3 Opik

Formerly CometLLM. Debug, evaluate, and monitor LLM applications with tracing and dashboards.

 Machine learning 22.4K 4,323 Apache-2.0
4 Phoenix

ML observability in a notebook with UMAP visualizations

 Machine learning 11.7K 3,725 Elastic-2.0
5 Rerun

An SDK for logging computer vision and robotics data paired with a visualizer for exploring that data over time.

 Python 11.5K 3,095 Apache-2.0
6 FiftyOne

Tool for visualizing, curating and evaluating computer vision datasets and models.

 Machine learning 11.1K 8,917 Apache-2.0
7 Visdom

Tool for real-time visualization and monitoring of live data such as ML experiments.

 Machine learning 10.3K 548 Apache-2.0
8 BertViz

Visualize attention in Transformer language models such as BERT and GPT-2.

 Machine learning 8.2K 1 Apache-2.0
9 TensorBoard

TensorFlow's visualization toolkit for metrics, model graphs, embeddings and more.

 Machine learning 7.2K 40 Apache-2.0
10 Aim

Experiment tracker with a UI to explore and compare ML runs and metrics.

 Machine learning 6.3K 1 Apache-2.0
11 Orange

Visual programming tool for data mining, machine learning and interactive data visualization.

 Apps & tools 5.7K 309 Other
12 Embedding Atlas

Interactive visualization of large embeddings with search, filtering and density views, by Apple.

 Machine learning 5K 134 MIT
13 Yellowbrick Inactive

Visual analysis and diagnostic tools for machine learning model selection with scikit-learn.

 Machine learning 4.4K 0 Apache-2.0
14 LIT Inactive

Learning Interpretability Tool: interactive visual analysis of ML model behavior, by Google PAIR.

 Machine learning 3.7K 0 Apache-2.0
15 TensorWatch

Debugging and visualization tool for data science and machine learning

 Machine learning 3.5K 6 MIT
16 dtreeviz

Decision tree visualization and model interpretation library for Python.

 Machine learning 3.2K 18 MIT
17 Comet

An MLOps platform for tracking, visualizing, and debugging your machine learning workflows from training straight through to production.

 Machine learning 175 38 MIT

Head-to-head comparisons

Categories in this topic

Frequently asked questions

What is the most popular ML visualization tool?

By GitHub stars, Netron is the most popular, with 33.5K stars, followed by PlotNeuralNet (25K) and Opik (22.4K).

Which ML visualization tools are actively maintained?

14 of the 17 tools listed had commits in the last 12 months, including Netron, Opik, Phoenix, Rerun, and FiftyOne. Each tool page shows monthly commit activity.

Are these ML visualization tools free and open source?

15 of the 17 tools listed use an OSI-approved open-source license; the others are source-available, free to use, or use a custom license, so check each tool page. The most common license here is Apache-2.0.

Missing a tool? Add it with a one-line pull request to the README.