jax

131ae7a4 - Merge pull request #30011 from jakevdp:fix-parallel-breakage