jax

[Testing do not merge] Add ROCm wheel build and test pipeline to continuous CI

#723

Open