jax

fix(ci): use HIP_VISIBLE_DEVICES count for pytest workers on DPX NPS2

#888

Open