Comment by N1H1L

There is also ucx-py that can be used with dask_cuda for rapid GPU-GPU communication

https://github.com/rapidsai/ucx-py