[FIX] Added jaxlib
version guard for CUDA compute capability check
#48080
Loading
jaxlib
version guard for CUDA compute capability check
#48080