softmax in a separate package