jax

23192072 - Merge pull request #30017 from jakevdp:finalize-xla-client-deps