jax

6587096a - Update XLA dependency to use revision http://github.com/openxla/xla/commit/fd900b9937238e19c57fd25bcdab200ea2e2f149