Numeric.AD.Variadic

Variadic combinators for variadic mixed-mode automatic differentiation.

Unfortunately, variadicity comes at the expense of being able to use quantification to avoid sensitivity confusion, so be careful when counting the number of lift you use when taking the gradient of a function that takes gradients!

Reverse-mode variadic gradient

class Num a => Grad i o o' a | i -> a o o', o -> a i o', o' -> a i o

Instances

Num a => Grad (AD Reverse a) [a] (a, [a]) a

Grad i o o' a => Grad (AD Reverse a -> i) (a -> o) (a -> o') a

vgrad :: Grad i o o' a => i -> o

vgrad' :: Grad i o o' a => i -> o'

Sparse forward mode variadic jet

class Num a => Grads i o a | i -> a o, o -> a i

Instances

Grads i o a => Grads (AD Sparse a -> i) (a -> o) a

Num a => Grads (AD Sparse a) (Cofree [] a) a

vgrads :: Grads i o a => i -> o