You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
{{ message }}
Repository navigation
[BUG] Complex vjps are wrong for cos, sqrt, and rsqrt (missing conjugate) #3780
mx.cos, mx.sqrt, and mx.rsqrt return the wrong gradient for complex inputs. cos's vjp delegates to its jvp, and sqrt/rsqrt form the derivative without conjugating it, so all three return f'(z) * cotangent instead of conj(f'(z)) * cotangent (the convention used by mx.exp/mx.log).
Activity
zcbenz commentedon Sep 26, 2026
Merging into #3778