I created Scorch: a minimalist neural net framework in Scala inspired by PyTorch.
You can find it here: https://github.com/botkop/scorch
Under the hood it uses the n-dimensional arrays of nd4j (https://nd4j.org/).
I am thinking of making a version that uses mxnet’s ndarray.
Do you think this makes sense?
(To be honest, I find the Scala interface of mxnet rather bloated.)