MLDataDevices.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)
- Package was renamed from
LuxDeviceUtils.jl
toMLDataDevices.jl
. Lux(***)Device
has been renamed to(***)Device
.Lux(***)Adaptor
objects have been removed. Use(***)Device
objects instead.