LuxDL
Replies: 1 comment
-
|
Currently the way the backend (LuxLib) is written, we do use polyester for CPU and KA for GPU. So there isn't anything performance wise to gain from using AK. That said, it will help us make the code much more concise and I have been meaning to try it out once I have some free time. |
Beta Was this translation helpful? Give feedback.
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
-
Would Lux.jl have anything to gain from https://github.com/anicusan/AcceleratedKernels.jl ?
Beta Was this translation helpful? Give feedback.
All reactions