jax

f5609fe4 - [TESTING ONLY] Bypass ROCm plugin version check and strip nightly workflow

[TESTING ONLY] Bypass ROCm plugin version check and strip nightly workflow This commit is for testing purposes only and should be removed before merge. Changes: 1. Add patch to bypass version check in jaxlib plugin_support.py for ROCm - Patch file: .github/workflows/patches/rocm_plugin_version_bypass.patch - Makes check_plugin_version() always return True for ROCm plugins 2. Apply patch in pytest_rocm.yml workflow after installing jaxlib - New step: "Apply ROCm plugin version bypass patch (TESTING ONLY)" - Finds installed jaxlib location and applies patch - Includes verification step to confirm patch was applied 3. Strip wheel_tests_nightly_release.yml to only run pytest-rocm job - Removed all jobs except run-pytest-rocm - Reduces CI time and allows focused testing - Updated workflow name to indicate testing mode Purpose: - Allows testing with mismatched ROCm plugin and jaxlib versions - Enables testing of new download workflow without version constraints - Speeds up CI by only running ROCm tests DO NOT MERGE - This is a temporary testing commit. Remove this commit before merging to main.