GPU execution
Run the sweep on NVIDIA or AMD GPUs through CuPy.
src_method runs on CPU (NumPy) by default and can be accelerated on GPUs
through CuPy. Both NVIDIA (CUDA) and AMD (ROCm) GPUs are
supported through optional extras:
uv pip install "src_method[gpu-nvidia]" # CUDA 13.x, driver >= 580
uv pip install "src_method[gpu-rocm]" # ROCmWhen adding src_method to another project's pyproject.toml:
dependencies = ["src_method[gpu-nvidia]"]At runtime, pass device="gpu":
result = apply(H, psi, chi_out=256, device="gpu")The inputs can be NumPy or CuPy arrays; they are moved to the device for the
sweep, and the result always comes back as NumPy arrays on the host. Without
CuPy installed, device="gpu" raises an ImportError.