jax

5089fb01 - Use new ML Build CUDA images

Use new ML Build CUDA images Note the CUDA jobs fail in https://github.com/jax-ml/jax/actions/runs/13444028636/job/37565647540 but in the same way as on HEAD. PiperOrigin-RevId: 729571786