jax version 0.3.13

Merge pull request jax-ml#10718 from hawkinsp:jaxfix

PiperOrigin-RevId: 448849099

Merge pull request jax-ml#10688 from sharadmv:color-by-default

PiperOrigin-RevId: 448723842

Update TF commit for release

PiperOrigin-RevId: 446555288

Update TF commit for release

PiperOrigin-RevId: 446555288

Enable a number of tests on GPU.

In particular, pjit/xmap work on CPU these days.

PiperOrigin-RevId: 446085110

jax version 0.3.8

Jaxlib release v0.3.7

Jax release v0.3.7

Upgrade the bazel version to 5.1.1

PiperOrigin-RevId: 441338363