Open avik-pal opened 1 week ago
batched_jacobian
Enzyme.Forward
Enzyme.Reverse
vector_jacobian_product
jacobian_vector_product
Enzyme needs us to use autodiff_thunk for the internal call, so we can easily replace the call inside custom rrules.
autodiff_thunk
EnzymeRules
CRC.rrule
Functions
batched_jacobianEnzyme.ForwardEnzyme.Reversevector_jacobian_productjacobian_vector_productNested Autodiff
Enzyme needs us to use
autodiff_thunkfor the internal call, so we can easily replace the call inside custom rrules.EnzymeRulesCRC.rrule