Code

Utilize ChainRulesCore thunks#966

Code

Merged

CarloLucibello merged 22 commits intoFluxML:masterFluxML/Zygote.jl:masterfrom oschulz:utilize-thunksoschulz/Zygote.jl:utilize-thunksCopy head branch name to clipboardJan 4, 2025

+82-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

  Copy full SHA for 09328d6

• Use @_adjoint_keepthunks where appropriate

  

  oschulz

  authored and

  pxl-th

  committed
  f91141a

  Copy full SHA for f91141a

• Use wrap_chainrules_output in unthunk_tangent

  

  oschulz

  authored and

  pxl-th

  committed
  1477280

  Copy full SHA for 1477280

• Fix unthunk_tangent for array of thunks

  

  oschulz

  authored and

  pxl-th

  committed
  4befdf5

  Copy full SHA for 4befdf5

• Don't unthunk explicitly in unbroadcast

  

  oschulz

  authored and

  pxl-th

  committed
  660cf4c

  Copy full SHA for 660cf4c

• Define unthunk_tangent for IdDict to support Params

  

  oschulz

  authored and

  pxl-th

  committed
  7938e04

  Copy full SHA for 7938e04

• Make unthunk_tangent for IdDict non-differentiable

  Show description for d6dec7d
  

  2 peopleauthored and

  pxl-th

  committed
  d6dec7d

  Copy full SHA for d6dec7d

• Revert "Don't unthunk explicitly in unbroadcast"

  Show description for 691c9ac
  

  oschulz

  authored and

  pxl-th

  committed
  691c9ac

  Copy full SHA for 691c9ac

• Fix problems related to unthunk_tangent for IdDict

  Show description for e8f0ce0
  

  2 peopleauthored and

  pxl-th

  committed
  e8f0ce0

  Copy full SHA for e8f0ce0

• Resolve duplicate rrule for unthunk_tangent with IdDict

  

  oschulz

  authored and

  pxl-th

  committed
  d0e106f

  Copy full SHA for d0e106f

• Make unthunk_tangent recurse into arrays

  

  oschulz

  authored and

  pxl-th

  committed
  f03a041

  Copy full SHA for f03a041

• Fix tests

  pxl-th

  committed
  b371c3a

  Copy full SHA for b371c3a

• Unthunk in collect(::Generator)

  pxl-th

  committed
  d621287

  Copy full SHA for d621287

• Update deps

  pxl-th

  committed
  e9c7805

  Copy full SHA for e9c7805

Commits on Jan 2, 2025

• Disable thunks for 2nd order AD

  pxl-th

  committed
  dcb3eb0

  Copy full SHA for dcb3eb0

• Temporary use fork of CRC

  pxl-th

  committed
  90c4d6e

  Copy full SHA for 90c4d6e

• Remove hook

  pxl-th

  committed
  a046f59

  Copy full SHA for a046f59

• Fix

  pxl-th

  committed
  fba8860

  Copy full SHA for fba8860

• Cleanup

  pxl-th

  committed
  8e6c2f5

  Copy full SHA for 8e6c2f5

Commits on Jan 3, 2025

• Up deps

  pxl-th

  committed
  f074ba2

  Copy full SHA for f074ba2

• Cleanup

  pxl-th

  committed
  d7ecca6

  Copy full SHA for d7ecca6

• Remove extra unthunk_tangent

  pxl-th

  committed
  2d21e2e

  Copy full SHA for 2d21e2e