jax

b11d86c8 - Update XLA dependency to use revision http://github.com/openxla/xla/commit/904baf4249be108fd349db443a313294dbbb7f69