Use the cuda 11.8 image for CPU builds to reduce the docker image churn from the JAX side in TF's RBE cluster
Use the cuda 11.8 image for CPU builds to reduce the docker image churn from the JAX side in TF's RBE cluster

PiperOrigin-RevId: 490522647

google/jax,google/jax,google/jax,google/jax