jax

b5010a2d - Merge pull request #22634 from jakevdp:array-api-cleanup