LuxAMDGPU.jl

Trigger Package for AMDGPU Support in Lux.jl
Author LuxDL
Popularity
3 Stars
Updated Last
5 Months Ago
Started In
March 2023

LuxAMDGPU

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

CI Buildkite NVIDIA GPU CI codecov Package Downloads

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

Caution

LuxAMDPU.jl was used previously as a means to load AMDGPU specific extensions in Lux.jl. However in most recent versions of Lux.jl, these functionalities can be directly accessed by simply loading AMDGPU.jl. This package is now deprecated.

To use the LuxAMDGPU.functional() functionality, use MLDataDevices.functional(LuxAMDGPUDevice).

If you are not using LuxAMDGPU.jl as a direct dependency, please consider opening an issue on any packages you are using that do use it as a dependency. From Julia 1.9 onwards, you can query ]why LuxAMDGPU to figure out which package originally brings it in as a dependency.

LuxAMDGPU is meant to be used as a trigger package for all AMDGPU dependencies in Lux. Users requiring AMDGPU support should install LuxAMDGPU and load it alongside Lux.

Installation

LuxAMDGPU is a registered package, and can be installed using the Julia package manager:

] add LuxAMDGPU

Used By Packages

No packages found.