jax

542cb2e5 - Fix a bug in jax.scipy.stats.rankdata leading to breakage with shape polymorphism.