jax

0ffa1103 - Update XLA dependency to use revision http://github.com/openxla/xla/commit/a3f650b2c88d1ee63191acf6ecb20920b7d933f8