I’m new to Zygote.jl and I’d like to compute the squared sum of a gradient, i.e., g = gradient(f, x), q = gradient(g’*g, x) Sucessfully did this using Forward mode: function grad_g_sqr(f, d, x) function gs(x) a = ForwardDiff.gradient(f, x) return a' * a / 2 end gg = F…

Compute gradient of gradient norm using zygote

ToucheSir April 20, 2022, 11:48pm 10

1 Like

Topic		Replies	Views
Newbie: Gradient of a gradient performance in Zygote General Usage zygote	2	511	March 21, 2021
Gradient of Gradient in Zygote General Usage	2	2735	January 1, 2021
Nested AD with Lux etc Machine Learning ad	26	1270	May 1, 2024
Differentiating Jacobian-vector product for sliced score matching? Machine Learning flux , zygote , ad	18	535	June 29, 2023
Mutation error in Zygote hessian Machine Learning question , package	2	184	June 22, 2023