jax

00930e85 - Update XLA dependency to use revision http://github.com/openxla/xla/commit/301198e60a88c879842588263fac1eed5dbf3a04