The Scala community has some clever techniques to ensure that those properties can be proven at compile-time, for example, https://github.com/eaplatanios/tensor

The Scala community has some clever techniques to ensure that those properties can be proven at compile-time, for example, https://github.com/eaplatanios/tensorflow_scala https://github.com/eaplatanios/tensorflow_scala for TensorFlow and https://github.com/ctongfei/nexus https://github.com/ctongfei/nexus