Added support for Softmax neural layers