jax

4a3bc901 - Merge pull request #4975 from jakevdp:reduction-initial