jax

6469f170 - Update XLA dependency to use revision http://github.com/openxla/xla/commit/66dbbf501ffd74f83c6a5d8fc201c756b1198d64