At some point during computation, my model needs to compute the logarithm of softplus of a parameter.
Currently, I implement this via:
torch.nn.functional.softplus(theta).log()

Due to the log() call I fear that there might be issues with computational stability.
Is there a computationally more stable way of computing log-softplus?
I do not want to change my parameterization.
Thank you!