(Feature Request) JupyterLab-compatible visualizations

I've been a long-time advocate of JupyterLab over plain Jupyter, but this is becoming even more relevant with the growing & maturing of SageMaker Studio's features since 2019...

Is any work being done on making the graph renders compatible with JupyterLab?

I see some libraries e.g. [ipydagred3](https://github.com/timkpaine/ipydagred3) that seem to suggest potentially re-usable patterns, but not sure how much focus there'd need to be on retaining the Step Functions console look & feel for accepting visualizations into this SDK.

I'd really like to be able to visualize in-progress executions from my (these days mostly SMStudio-based) notebook workflows.