Skip to content

Latest commit

 

History

History
23 lines (17 loc) · 1.81 KB

README.md

File metadata and controls

23 lines (17 loc) · 1.81 KB

DeviceUtils

Join the chat at https://julialang.zulipchat.com #machine-learning Latest Docs Stable Docs

CI Buildkite codecov Aqua QA

ColPrac: Contributor's Guide on Collaborative Practices for Community Packages SciML Code Style

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:

  1. CUDA.jl for NVIDIA GPUs.
  2. AMDGPU.jl for AMD ROCM GPUs.
  3. Metal.jl for Apple Metal GPUs. (Experimental)
  4. oneAPI.jl for Intel GPUs. (Experimental)