Call _check_arraylike in jax.numpy to improve error messages
This PR only adds the call to places where non-array inputs currently lead to errors. There
remain a number of other functions where adding this check would lead to potentially breaking
changes; these are deliberately left out of this PR.
