Add tests for inplace Sigmoid and Softplus

Also adds an inplace version
And fixed the jax code to be equivalent to the Aesara implementation

In ce9c17f3 the RV was changed such that `p` can be multi-dimensional instead of just 1-dimensional.
Since then the support length is determined by the last dimension of `p`.