jax

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

#792

Merged