ProjectionLayer and Softmax improvements