follow up to #363 we might need to do the same trick as in https://github.com/JuliaDiff/ChainRulesCore.jl/pull/308 for the `rrule` with `RuleConfig`.
