jax

92ebb533 - Merge pull request #22181 from jakevdp:xla-abbrevs