https://blog.ando.ai/posts/softmax-to-the-max/