Try Cuda 11.2 build in presubmit for GPU. Presubmit uses 11.2 and CI uses 11.1
Try Cuda 11.2 build in presubmit for GPU. Presubmit uses 11.2 and CI uses 11.1

PiperOrigin-RevId: 473071603

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