DeviceUtils.jl
is a lightweight package defining rules for transferring data across
devices. It is used in deep learning frameworks such as Lux.jl.
Currently we provide support for the following backends:
CUDA.jl
for NVIDIA GPUs.AMDGPU.jl
for AMD ROCM GPUs.Metal.jl
for Apple Metal GPUs. (Experimental)oneAPI.jl
for Intel GPUs. (Experimental)