jax

8d9e1809 - Update XLA dependency to use revision http://github.com/openxla/xla/commit/11b961377ccfb82d0b0256c9852c0042a67e520e