Discrepancy between complex gradients calculate with Zygote.jl and Python's Jax

The solutions are conjugates of each other.
If I remember rightly, whether a gradient should return a jacobian transpose, or a jacobian adjoint (i.e. conjugate transpose) is a matter of debate/convention.

in 2020 @Mason did a deep dive on the topic:

including Wirtinger Derivatives.
Our convention come out of that discussion.

2 Likes