Code

Utilize ChainRulesCore thunks#966

Code

Open

oschulz wants to merge 14 commits intoFluxML:masterFluxML/Zygote.jl:masterfrom oschulz:utilize-thunksoschulz/Zygote.jl:utilize-thunksCopy head branch name to clipboard

+86-42

Commits

Commits on Jan 1, 2025

• Don't force unthunking of ChainRulesCore thunks

  Show description for 09328d6
  

  oschulz

  authored and

  pxl-th

  committed
  09328d6

• Use @_adjoint_keepthunks where appropriate

  

  oschulz

  authored and

  pxl-th

  committed
  f91141a

• Use wrap_chainrules_output in unthunk_tangent

  

  oschulz

  authored and

  pxl-th

  committed
  1477280

• Fix unthunk_tangent for array of thunks

  

  oschulz

  authored and

  pxl-th

  committed
  4befdf5

• Don't unthunk explicitly in unbroadcast

  

  oschulz

  authored and

  pxl-th

  committed
  660cf4c

• Define unthunk_tangent for IdDict to support Params

  

  oschulz

  authored and

  pxl-th

  committed
  7938e04

• Make unthunk_tangent for IdDict non-differentiable

  Show description for d6dec7d
  

  2 peopleauthored and

  pxl-th

  committed
  d6dec7d

• Revert "Don't unthunk explicitly in unbroadcast"

  Show description for 691c9ac
  

  oschulz

  authored and

  pxl-th

  committed
  691c9ac

• Fix problems related to unthunk_tangent for IdDict

  Show description for e8f0ce0
  

  2 peopleauthored and

  pxl-th

  committed
  e8f0ce0

• Resolve duplicate rrule for unthunk_tangent with IdDict

  

  oschulz

  authored and

  pxl-th

  committed
  d0e106f

• Make unthunk_tangent recurse into arrays

  

  oschulz

  authored and

  pxl-th

  committed
  f03a041

• Fix tests

  pxl-th

  committed
  b371c3a

• Unthunk in collect(::Generator)

  pxl-th

  committed
  d621287

• Update deps

  pxl-th

  committed
  4795f7a