jax

d1a8f5c9 - Update XLA dependency to use revision http://github.com/openxla/xla/commit/7e3a637c7f11ab7afe3da16f54539aefdae4aa47