Switch autograd to ChainRules.jl
instead of using Zygote.jl
adjoints
#94
Labels
enhancement
New feature or request
ChainRules.jl
instead of using Zygote.jl
adjoints
#94
Writing the adjoints with
ChainRules.rrule
will allow the RENs to be used by other auto-grad engines than Zygote.The text was updated successfully, but these errors were encountered: