Remove dynamic discovery of unknown rocm wheels.
Calling pkgutil.iter_modules() can be very slow (seconds) if you have many, but this entire mechanism is not really needed. We don't need to support unknown ROCM versions with old JAX releases. PiperOrigin-RevId: 969369379
P
Peter Hawkins committed
d56fa6aef77a0e6612b4076f132175f9c562076b
Parent: eecec8d
Committed by jax authors <google-ml-automation@google.com>
on 8/23/2026, 1:24:56 PM