Convenience: cbrt and atan2.
Author: mcourteauxCreated Jan 15, 2026Updated Jan 15, 2026
The cubic root gradient is not implemented directly. We have to resort to sign(a) * power(abs(a), 1/3) for now. Also, arctan2 is defined with a gradient, but the same function atan2, which is basically an alias is not defined in autograd.
Source: HIPS/autograd