Currently NVIDIA-only, we're cooking up some Vulkan stuff in rust-gpu though.
In years prior I wouldn't have even bothered, but it's 2026 and AMD's drivers actually come with a recent version of torch that 'just works' on windows. Anything is possible :)