JuliaSmoothOptimizers / LinearOperators.jl

Linear Operators for Julia
Other
150 stars 32 forks source link

Chainrules #278

Closed migrosser closed 1 year ago

migrosser commented 1 year ago

It would be very nice to implement ChainRulesCore.frule and ChainRulesCore.rrule for products of an LinearOperator with a vector. This should make it quite straight-forward to use LinearOperators in conjuction with Flux.jl / Zygote.jl. In particular it would remove the need to implement custom chainrules when the implementation of the product is not in a form suitable for Zygote.