jax

ae3d3101 - Merge pull request #4524 from jakevdp:fix-mean