Upgrade from numpy to cupy (https://cupy.dev/) Detect if there is GPU --> if true, use cupy
Upgrade from numpy to cupy (https://cupy.dev/)
Detect if there is GPU --> if true, use cupy