Nextjournal / Aug 03 2019

Clojure MXNet

Run Clojure and MXNet on the GPU.

(require '[org.apache.clojure-mxnet.ndarray :as ndarray]
         '[org.apache.clojure-mxnet.context :as context])

(ndarray/->vec (ndarray/+
 (ndarray/zeros [3 3] {:ctx (context/gpu 0)})
 (ndarray/ones [3 3] {:ctx (context/gpu 0)})))
Vector(9) [1, 1, 1, 1, 1, 1, 1, 1, 1]