Softmax Numerical Stability & Log-Sum-Exp | TensorTonic