NumPy-CuPy Generic Code SupportΒΆ

cupy.get_array_module Returns the array module for arguments.
cupyx.scipy.get_array_module Returns the array module for arguments.