jax

affa2dcc - Increment jax and jaxlib versions. (#1603)

Increment jax and jaxlib versions. (#1603) * Update XLA version to https://github.com/tensorflow/tensorflow/commit/7acd3bb9d794e279c6d824f5553fee2703c16f35 * Remove XRT reference from jaxlib build.