Softmax Numerical Stability & Log-Sum-Exp

Implement both naive softmax and the numerically stable version using the log-sum-exp trick.