Build fails CI/CD tests with message like RuntimeError: PyTorch is not linked with support for jax devices on "GPU tests requiring torch CUDA / test (python_tests, linux.8xlarge.nvidia.gpu)"

[yaoshiang]

it may have started with PR #9156 on ffd4488