CuPy is an implementation of a NumPy-compatible multi-dimensional array on CUDA.
conda install nsls2forge::cupy