Native implementation of Deep Learning models for GPU backends (mxnet, Caffe, TensorFlow, etc.)
sudo apt-get install libatlas-base-dev libblas-dev libopencv-dev swig -y
git clone https://github.com/h2oai/deepwater.git
git submodule update --init --recursive
cd mxnet
make
cd ..
make
make pkg # will generate a `jar` file including native code.
Please add these lines below into your Java code when using this jar as external dependency.
util.loadCudaLib();
util.loadNativeLib("mxnet");
util.loadNativeLib("Native");
An example implementation can be found in H2O.