Any plan to support GPU?

[whbldhwj]

Hi,

This looks like a really interesting project! I saw currently torcharrow runs on CPU with the Velox backend. Just wondering any plan to offload some of the ops to GPU? If exists, what's the technical path? Any plan to reuse the cudf/rapids library?

Thanks,
Jie

[dracifer]

Thanks @whbldhwj ! we do plan on supporting GPU. Current prototype is based on PyTorch itself. Would you mind sharing if have particular use cases that can benefit from GPU processing?