Skip to content

mlflow-widgets

Anywidget-based notebook widgets for MLflow experiment visualization.

Inspired by wigglystuff's WandbChart, but built for MLflow.

Overview

mlflow-widgets provides interactive, live-updating widgets that work in Jupyter, Marimo, and any notebook environment that supports anywidget.

Widget Description
MlflowChart Live-updating canvas line chart for MLflow metrics
MlflowRunTable Sortable HTML table of experiment runs with nested run tree
MlflowParallelCoordinates Parallel coordinates chart for hyperparameter comparison
MlflowRunSelector Pure Python helper returning pandas DataFrames for run selection

Key features

  • Live-updating canvas-based line chart with multiple x-axis modes (step, wall time, relative time)
  • Interactive clickable legend with per-series toggle and nested run grouping
  • Experiment run table with sortable columns and nested run tree view
  • Parallel coordinates chart with parameter/metric filtering and multiple color modes
  • Smoothing controls: rolling mean, exponential moving average, gaussian
  • Hover tooltips with exact values
  • Auto-polling with configurable interval
  • Works in Jupyter, Marimo, and any notebook that supports anywidget

Quick example

from mlflow_widgets import MlflowChart

chart = MlflowChart(
    tracking_uri="http://localhost:5000",
    experiment_id="1",
    metric_key="loss",
)
chart

For AI assistants

This site exposes LLM-friendly endpoints:

Every page is also available as raw Markdown at the same URL path with an .md extension (e.g. /installation/index.md).