jax

af29a7b0 - Merge pull request #26372 from jakevdp:split-lax-numpy