Can not install jaxlib==0.3.24 under cuda 12 environment

[erwushth]

Hi,
When I tried to install RFDesign, get pip subprocess error:
ERROR: Could not find a version that satisfies the requirement jaxlib==0.3.24 (from versions: 0.1.63, 0.4.0, 0.4.1, 0.4.2, 0.4.3, 0.4.4, 0.4.6, 0.4.7, 0.4.9, 0.4.10, 0.4.11, 0.4.12, 0.4.13, 0.4.14, 0.4.15, 0.4.16, 0.4.17, 0.4.18, 0.4.19, 0.4.20) ERROR: No matching distribution found for jaxlib==0.3.24

I confirmed that the issue was caused by installing jaxlib==0.3.24 under cuda 12 environment, since the version of jaxlib was built in cuda 11 environment.

Will it cause compatibility issues, if I install jaxlib==0.4.0？

[rcrehuet]

The installation worked for me with jaxlib==0.4. However, I get an execution error when I run the tests, which I have reported in this issue #58.