Comment by hatch_q

No. Sadly, NVIDIA killed any kind of compute via Vulkan.

I took few minutes to try to make it work on ROCm (AMD's alternative to CUDA), landed in python dependency hell.