Using Multiple GPUs in build_transform

[dyyjenny]

Hi, I have a relatively large paired ST and H&E image, where there are several pieces of tissue on one slice, and I run into cuda out of memory error when trying to run phi = STalign.build_transform(out['xv'],out['v'],out['A'],XJ=[YI,XI],direction='f'). Wondering if anyone has tried splitting data and running stalign on multiple gpus? Thank you!