MXNet - Scala API

MXNet supports the Scala programming language. The MXNet Scala package brings flexible and efficient GPU computing and state-of-art deep learning to Scala. It enables you to write seamless tensor/matrix computation with multiple GPUs in Scala. It also lets you construct and customize the state-of-art deep learning models in Scala, and apply them to tasks, such as image classification and data science challenges.

Image Classification with the Scala Infer API

The Infer API can be used for single and batch image classification. More information can be found at the following locations:

Tensor and Matrix Computations

You can perform tensor or matrix computation in pure Scala:

   import org.apache.mxnet._

   val arr = NDArray.ones(2, 3)
   // arr: org.apache.mxnet.NDArray = org.apache.mxnet.NDArray@f5e74790

   arr.shape
   // org.apache.mxnet.Shape = (2,3)

   (arr * 2).toArray
   // Array[Float] = Array(2.0, 2.0, 2.0, 2.0, 2.0, 2.0)

   (arr * 2).shape
   // org.apache.mxnet.Shape = (2,3)