jax

a721ae2a - Merge pull request #18923 from jakevdp:ci-setup-python