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/ones [3 3] {:ctx (context/gpu 0)})
 (ndarray/ones [3 3] {:ctx (context/gpu 0)})))

0.0s

Clojure

Vector(9) [2, 2, 2, 2, 2, 2, 2, 2, 2]

nvidia-smi

0.6s

Bash in Clojure