org.apache.spark.ml.ann.ANNSuite.ANN with SoftMax learns XOR function with 2-bit output and batch GD optimizer