https://blog.tnichols.org/posts/cross-entropy/
Deriving Cross Entropy Gradient - Tyler's Blog