LuxDL
Replies: 1 comment 3 replies
-
|
Zygote is the default AD rn only because it integrates well with chainrules and covers all the cases we need. We will indeed be moving towards Enzyme.jl . FWIW I have tested the latest blas support of enzyme with lux and it seems to work on CPU for simple models. |
Beta Was this translation helpful? Give feedback.
-
|
That sounds great! Keep up the great work! ; ) I will try it out in my master's project. |
Beta Was this translation helpful? Give feedback.
-
|
Just spotted a typo |
Beta Was this translation helpful? Give feedback.
-
|
No that does link to the tutorial from flax |
Beta Was this translation helpful? Give feedback.
-
There are plenty of gradient error reports on Zygote.jl which worries people like me, Would you like to consider another much safer AD system? heard about Enzyme, but not sure if it is better.
Beta Was this translation helpful? Give feedback.
All reactions