jax

bb0ce4e8 - Update XLA dependency to use revision http://github.com/openxla/xla/commit/9e665ebe7b6fad3bef7147bc79f65440d19bfc58