jax

f056bd87 - Update XLA dependency to use revision http://github.com/openxla/xla/commit/0ef91e2443e93d9d4649c1a73bdd71f95319b12c