jax

a1747e18 - Merge pull request #36292 from jakevdp:argsort-stable