Softmax and Numerically Stable Cross-Entropy | AlgoML Prep