jax

5a780a03 - Update XLA dependency to use revision http://github.com/openxla/xla/commit/77674103aaa978d6b8f50b777ede7d5a8686347a