udate nnff_gpu with num stable softmax