jax

7ed640bd - Bump pytest workers to 8/GPU, add bazel-skip toggle for nightly dispatch

Bump pytest workers to 8/GPU, add bazel-skip toggle for nightly dispatch - rocm_test_env.sh: bump xdist workers from 2 to 8 per GPU. - wheel_tests_nightly_release.yml: add skip-bazel-rocm-tests dispatch input (default '0') so a manual dispatch can run pytest only, without spending the gfx950.1 pool on run-bazel-test-rocm.