jax

89c7403d - Merge pull request #27376 from jakevdp:jtu-type-annotations