Matthew Rocklin's post on first stps towards mixing dask with GPU arrays: http://matthewrocklin.com/blog/work/2019/01/03/dask-array-gpus-first-steps
This video also looks interesting: 'Scaling Up and Out Programming GPU Clusters with Numba and Dask | SciPy 2016 | Siu Kwan Lam' https://www.youtube.com/watch?v=jLQ2wAwapRg
GitHub repo: imreg_dft Image registration using discrete Fourier transform. https://github.com/matejak/imreg_dft
More python image registration: https://github.com/pyimreg