softmax in a seperate package