jax

de7e49e1 - Update XLA dependency to use revision http://github.com/openxla/xla/commit/08ffe8376f88e04210b2e4624871cc75edb7baed