Bump TPU VM Jax to 0.4.23

http://b/322834705

Author: djherbis

tpu/config.txt

@@ -5,7 +5,7 @@ PYTHON_VERSION_PATH=python3.10
 TENSORFLOW_VERSION=2.15.0
 TF_LIBTPU_VERSION=1.9.0
 TF_LINUX_WHEEL_VERSION=manylinux_2_17_x86_64.manylinux2014_x86_64
-JAX_VERSION=0.4.17
+JAX_VERSION=0.4.23
 # Supports nightly
 TORCH_VERSION=2.1.0
 # https://github.com/pytorch/audio supports nightly
@@ -14,4 +14,4 @@ TORCHAUDIO_VERSION=2.1.0
 TORCHTEXT_VERSION=0.16.0
 # https://github.com/pytorch/vision supports nightly
 TORCHVISION_VERSION=0.16.0
-TORCH_LINUX_WHEEL_VERSION=manylinux_2_28_x86_64
+TORCH_LINUX_WHEEL_VERSION=manylinux_2_28_x86_64