Inefficient Hessian matrix computation of function with finite difference JVP rule
Opened this issue · 0 comments
wcxve commented
This inefficiency occurs when some parameters are fixed during computation. The finite differences will be computed for these fixed parameters, which are not (?) optimized by JAX.