Handle transposition of arbitrary custom JVPs
Opened this issue · 0 comments
samestep commented
Similar to #76, as of #105 we only support transposition of functions that either are automatically derived JVPs or are very simple functions from a few F64 parameters returning an F64. We should allow transposition of other functions that deal with purely nonlinear or purely linear data, and maybe even more complicated mixtures of the two.