jax

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

#794

Merged