jax

786079b7 - Update XLA dependency to use revision http://github.com/openxla/xla/commit/500d7b12672ca2377aa84f42fad18a7b6f89e2aa