We hope that our NIFTS library can support alternative backends such as cupy, dask, jax.numpy, etc. Some of the source code already reflects a flexibility, but as of now, this feature has not been tested.
If you are pursuing this implementation or want to follow the progress, please make yourself known here.