Skip to content

Commit

Permalink
Merge pull request #7 from avik-pal/patch-1
Browse files Browse the repository at this point in the history
chore: don't load LuxDeviceUtils (deprecated)
  • Loading branch information
vpuri3 authored Nov 14, 2024
2 parents 26a4697 + 0c24bc9 commit d009a9d
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ y = layer(x, p, st)
We compare the performance of different implementation of KAN with an MLP that has the same number of parameters (see `examples/eg1.jl`).
```julia
using Lux, KolmogorovArnold
using CUDA, LuxDeviceUtils
using LuxCUDA

CUDA.allowscalar(false)
device = Lux.gpu_device()
Expand Down

0 comments on commit d009a9d

Please sign in to comment.