Note that they used a pretty old+forked version of keras for the mxnet backend