jax

cf21f733 - Merge pull request #27258 from jakevdp:fix-lint