Comment by ghostly_s

TFA mentions this project has backends for :

AMD GPU (AMDGPU.jl) NVIDIA GPU (CUDA.jl) MacBook GPU(Metal.jl)