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.jlfor NVIDIA GPUs.AMDGPU.jlfor AMD ROCM GPUs.Metal.jlfor Apple Metal GPUs. (Experimental)oneAPI.jlfor Intel GPUs. (Experimental)
- Package was renamed from
LuxDeviceUtils.jltoMLDataDevices.jl. Lux(***)Devicehas been renamed to(***)Device.Lux(***)Adaptorobjects have been removed. Use(***)Deviceobjects instead.