LuxDL
status of enzyme
#406
Replies: 2 comments
-
|
There is no timeline for now. But there is some upstream activity to make Enzyme compatible with SciMLSensitivity.jl which by extension will make Lux compatible with Enzyme. FWIW most simple things already work, but things that remain are the chain rules core rules for LuxLib and NNlib that need to be migrated. |
Beta Was this translation helpful? Give feedback.
0 replies
-
|
Enzyme is now the recommended backend for CPU models https://lux.csail.mit.edu/stable/manual/autodiff#autodiff-recommendations. GPU support is underway in Enzyme.jl repo |
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
-
It is mentioned in the documentation that there is a plan to move towards Enzyme.
I would like to learn if there is an estimated time for this migration and what is the current status.
Thanks in advance : )
Beta Was this translation helpful? Give feedback.
All reactions