fully functional softmax reg