I think you intended this to be outside the softplus. The reason is that it should be applied just before the log to avoid `log(0)=inf`.e.g.
- `log(softplus(-1000+1e-5))=log(0)=inf`.
- `log(softplus(-1000)+1e-5)=log(1e-5)!=inf`.
Also this fixes a NaN I had.
