I've fixed all of the other issues except for this one.

What would the `rescale_softmax` function look like? I'm a bit reluctant to modify the plushmax code because I had many `NaN`s when I used softmax instead of my plushmax function and I'd like to avoid those problems again.