Skip to content

CarloLucibello/LuxDeviceUtils.jl

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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)

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Packages

No packages published

Languages

  • Julia 99.1%
  • Shell 0.9%