move to stable version of nd4j and get rid of mallet dep