jax

ce621cf5 - PR #35516: [ROCm] Include prebuilt rocm plugin pkgs as deps to tests

PR #35516: [ROCm] Include prebuilt rocm plugin pkgs as deps to tests Imported from GitHub PR https://github.com/jax-ml/jax/pull/35516 Use pre-built (released version) of the rocm plugin to test the jax adaptions Copybara import of the project: -- 382ce0fe24041a65a4c10a9bfe5249e524812bc4 by Alexandros Theodoridis <atheodor@amd.com>: Include prebuilt rocm plugin pkgs as deps to tests Merging this change closes #35516 COPYBARA_INTEGRATE_REVIEW=https://github.com/jax-ml/jax/pull/35516 from ROCm:implement_rocm_plugin_release_pipeline_main 382ce0fe24041a65a4c10a9bfe5249e524812bc4 PiperOrigin-RevId: 884868946