SoftTreeMax Policy Gradient algorithm from https://arxiv.org/pdf/2301.13236.pdf - View it on GitHub