RNN distributed training with Spark and DL4J