import mxnet as mx