jax

4febce00 - [ROCm] Docker files cleanup

[ROCm] Docker files cleanup - Consolidate some of the changes to wheel copies - Fix inconsistency issues with gpu targets - sqlite3 and bz2 deps are only need for pyenv containers