jax

[ROCm] xdist: select GPU from runner-allocated ROCR slice

#793

Open