install mxnet

sudo pip3 install mxnet==0.11.0

import mxnet as mx

a = mx.nd.ones((2, 3))
b = a * 2 + 1
print(b.asnumpy())

result:

[[ 3.  3.  3.]
 [ 3.  3.  3.]]