jax

fc393eca - Update XLA dependency to use revision http://github.com/openxla/xla/commit/1418877d9b78f6e85c4786ee8427d52b8b7c9fe6