TowerOfEnzyme
API
TowerOfEnzyme.nth_derivative
— Functionnth_derivative(f, x, Val(Order))
nth_derivative(f, x, dx, Val(Order))
Calculates the Order
-th directional derivative of f
with regards to x
in direction dx
. If x
is a scalar dx
can be omitted and is set to one(x)
.
TowerOfEnzyme.derivative_bundle
— Functionderivative_bundle(f, (x, dx, ddx...))
Calculates the higher-order pushforward.
Helper functions
TowerOfEnzyme.jvp
— Functionjvp(f, x, v)
Calculate the Jacobian-Vector-Product $J_f(x) * v$
TowerOfEnzyme.hvvp
— Functionhvvp(f, x, v)
Calculate the Hessian-Vector-Vector product $H_f(x) * v * v$