MXNet - Scala API

See the MXNet Scala API Documentation.

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.

You can perform tensor or matrix computation in pure Scala:

   scala> import ml.dmlc.mxnet._
   import ml.dmlc.mxnet._

   scala> val arr = NDArray.ones(2, 3)
   arr: ml.dmlc.mxnet.NDArray = ml.dmlc.mxnet.NDArray@f5e74790

   scala> arr.shape
   res0: ml.dmlc.mxnet.Shape = (2,3)

   scala> (arr * 2).toArray
   res2: Array[Float] = Array(2.0, 2.0, 2.0, 2.0, 2.0, 2.0)

   scala> (arr * 2).shape
   res3: ml.dmlc.mxnet.Shape = (2,3)

Scala API Reference

  • Module API is a flexible high-level interface for training neural networks.
  • Model API is an alternate simple high-level interface for training neural networks.
  • Symbolic API performs operations on NDArrays to assemble neural networks from layers.
  • IO Data Loading API performs parsing and data loading.
  • NDArray API performs vector/matrix/tensor operations.
  • KVStore API performs multi-GPU and multi-host distributed training.