GPU acceleration not possible: BasicPy pinned to JAX ≤0.4.23, but GPU JAX wheels require JAX ≥0.6

[lauraccch]

Problem Description: BasicPy 1.2.0 requires JAX ≤0.4.23, but GPU-enabled JAX wheels (jax[cuda12]/cuda13) are only available for JAX ≥0.6.x. This prevents BasicPy from using GPU acceleration on CUDA-capable systems.

To reproduce:

mamba create -n basicpy_test python=3.10
conda activate basicpy_test
pip install basicpy
pip install --upgrade "jax[cuda12]"

Error Message after installing "jax[cuda12]":

Installing collected packages: jax-cuda12-pjrt, nvidia-nvjitlink-cu12, nvidia-nccl-cu12, nvidia-cuda-runtime-cu12, nvidia-cuda-nvrtc-cu12, nvidia-cuda-nvcc-cu12, nvidia-cuda-cupti-cu12, nvidia-cuda-cccl-cu12, nvidia-cublas-cu12, jax-cuda12-plugin, nvidia-nvshmem-cu12, nvidia-cusparse-cu12, nvidia-cufft-cu12, nvidia-cudnn-cu12, jaxlib, nvidia-cusolver-cu12, jax
  Attempting uninstall: jaxlib
    Found existing installation: jaxlib 0.4.23
    Uninstalling jaxlib-0.4.23:
      Successfully uninstalled jaxlib-0.4.23
  Attempting uninstall: jax
    Found existing installation: jax 0.4.23
    Uninstalling jax-0.4.23:
      Successfully uninstalled jax-0.4.23
ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.
basicpy 1.2.0 requires jax<=0.4.23,>=0.3.10, but you have jax 0.6.2 which is incompatible.
basicpy 1.2.0 requires jaxlib<=0.4.23,>=0.3.10, but you have jaxlib 0.6.2 which is incompatible.
Successfully installed jax-0.6.2 jax-cuda12-pjrt-0.6.2 jax-cuda12-plugin-0.6.2 jaxlib-0.6.2 nvidia-cublas-cu12-12.9.1.4 nvidia-cuda-cccl-cu12-12.9.27 nvidia-cuda-cupti-cu12-12.9.79 nvidia-cuda-nvcc-cu12-12.9.86 nvidia-cuda-nvrtc-cu12-12.9.86 nvidia-cuda-runtime-cu12-12.9.79 nvidia-cudnn-cu12-9.19.0.56 nvidia-cufft-cu12-11.4.1.4 nvidia-cusolver-cu12-11.7.5.82 nvidia-cusparse-cu12-12.5.10.65 nvidia-nccl-cu12-2.29.3 nvidia-nvjitlink-cu12-12.9.86 nvidia-nvshmem-cu12-3.5.21