Sarem Seitz

Would be very helpful if this was implemented.

Thank you, that would probably explain it. My understanding was that a ChainRule is applied equivalently in all autodiff packages, but that must have been wrong then. Do you recommend...