update nnff with num stable softmax