Fix `is_jax_array` for `jax>=0.8.2`#369

amacati wants to merge 4 commits intodata-apis:maindata-apis/array-api-compat:mainfrom

amacati:mainamacati/array-api-compat:mainCopy head branch name to clipboard