It should be relatively easy to support CuPy in addition to NumPy. See e.g. https://github.com/lab-cosmo/mops/pull/37