Most don’t use an eigendecomposition.
It’s probably easiest to differentiate the ExponentialUtiltiies.jl implementation, but the missing part will be Lapack balance.
We overlay the Enzyme.autodiff function, so Reactant is not using the mechanism in Enzyme.jl but directly calling the Enzyme C-function and only the MLIR part as of right now. This means that currently you can’t define custom frules/rrules; we only diff the MLIR that we can trace.
This sth we want to fix in the near/mid-term.
Ok, I understand. Thanks for the explanation