jax

593afa67 - Merge pull request #22663 from jakevdp:array-api-cleanup