Utilize ChainRulesCore thunks#966

Merged

CarloLucibello merged 22 commits intoFluxML:masterfrom

oschulz:utilize-thunks

Jan 4, 2025

Commits on Jan 1, 2025

Don't force unthunking of ChainRulesCore thunks

oschulz
authored and
pxl-th
committed
Use @_adjoint_keepthunks where appropriate

oschulz
authored and
pxl-th
committed
Use wrap_chainrules_output in unthunk_tangent

oschulz
authored and
pxl-th
committed
Fix unthunk_tangent for array of thunks

oschulz
authored and
pxl-th
committed
Don't unthunk explicitly in unbroadcast

oschulz
authored and
pxl-th
committed
Define unthunk_tangent for IdDict to support Params

oschulz
authored and
pxl-th
committed
Make unthunk_tangent for IdDict non-differentiable

authored and
pxl-th
committed
Revert "Don't unthunk explicitly in unbroadcast"

oschulz
authored and
pxl-th
committed
Fix problems related to unthunk_tangent for IdDict

authored and
pxl-th
committed
Resolve duplicate rrule for unthunk_tangent with IdDict

oschulz
authored and
pxl-th
committed
Make unthunk_tangent recurse into arrays

oschulz
authored and
pxl-th
committed
Fix tests
pxl-th
committed
Unthunk in collect(::Generator)
pxl-th
committed
Update deps
pxl-th
committed

Commits on Jan 2, 2025

Commits on Jan 3, 2025