jax

a71f9a62 - Merge pull request #25271 from jakevdp:fix-vector-norm