jax

0c10daed - Merge pull request #28661 from jakevdp:jax-array-jep