JupyterLab extension for learning traces