jax

1743f2c4 - Merge pull request #25371 from jakevdp:mypy-numpy-version