jax

3d049473 - relax version requirements for ROCm Jax Plugin wheels, allow overriding using env var