Change ROCm setup for plugin

Author: Ruturaj4

setup.py

@@ -107,7 +107,7 @@ def load_version_module(pkg_path):
         # ROCm support for ROCm 6.0 and above.
         'rocm': [
           f"jaxlib=={_current_jaxlib_version}",
-          f"jax-rocm60-plugin>={_current_jaxlib_version},<={_jax_version}",
+          "jax-rocm60-plugin>=0.5.0",
         ],
 
         # For automatic bootstrapping distributed jobs in Kubernetes