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:
CPUDevice
: for CPUs -- no additional packages required.CUDADevice
: CUDA.jl
for NVIDIA GPUs.AMDGPUDevice
: AMDGPU.jl
for AMD ROCM GPUs.MetalDevice
: Metal.jl
for Apple Metal GPUs. (Experimental)oneAPIDevice
: oneAPI.jl
for Intel GPUs. (Experimental)XLADevice
: Reactant.jl
for XLA Support. (Experimental)LuxDeviceUtils.jl
to MLDataDevices.jl
.Lux(***)Device
has been renamed to (***)Device
.Lux(***)Adaptor
objects have been removed. Use (***)Device
objects instead.