Issue Description

The UI configuration/set-up for ND4J (see link below) does not seem to work for SameDiff. I could not find anywhere in the documentation or code examples/tests that show me how to set-up visualization when using SameDiff.

The setListeners function in SameDiff dores not support a StatsListener listener as argument.

I did look at the UIListenerTest.java and TestSameDiffUI.java. They were not much help.

What do you suggest I do next?

Thanks

https://deeplearning4j.konduit.ai/deeplearning4j/how-to-guides/tuning-and-training/visualization