[WIP] Use cupy, in which case all operations are performed on GPU
#1051
| Job | Run time |
|---|---|
| 2m 36s | |
| 2m 36s |
cupy, in which case all operations are performed on GPU
#1051
| Job | Run time |
|---|---|
| 2m 36s | |
| 2m 36s |