Implement a Numerically Stable Softmax | AlgoML Prep