ROCM HIP / Vulkan support for GPU acceleration

**Is your feature request related to a problem? Please describe:**
Right now, it is not possible to use this library on AMD hardware because it is ABI-linked to either the CPU or CUDA version of Torch. It doesn't appear to be possible to build it with just replacing the torch version with rocm-enabled torch unlike what usually works for other ML projects.

**Describe the solution you would like:**
Be able to have rocm enabled build of this library available or at least implement HIP/Vulkan support so that it can be built from source.

**Describe the alternatives you have considered:**
I have none at this moment besides running on the CPU.

**Additional Context:**
ROCM 5.6.x / Radeon 7900 XTX working with most other tools relying on Torch.