invenia / Nabla.jl

A operator overloading, tape-based, reverse-mode AD
Other
67 stars 5 forks source link

Support for configured rrules. #201

Open oxinabox opened 3 years ago

oxinabox commented 3 years ago

Follow up from #189

Right now we don't support any ChainRules rrules that need a RuleConfig (see https://juliadiff.org/ChainRulesCore.jl/dev/config.html) These are for higher order functions.

We actually have a ton of them already hand-coded for Nabla, that use ForwardDiff that would ideally be good to move to ChainRules.jl.

Right now ChainRulesOverloadGeneration won't even tell us about them. Would need to solve https://github.com/JuliaDiff/ChainRulesOverloadGeneration.jl/issues/7 first.