implementation("org.jetbrains.kotlinx:kotlin-deeplearning-visualization:0.5.2")
This module contains the Kotlin API for visualization of the Deep Learning models built with the KotlinDL.