jax

a1dea76f - Merge pull request #34774 from magaonka-amd:enable-lobpcg-tests-rocm