Numeric.AD.Mode.Directed

Allows the choice of AD Mode to be specified at the term level for benchmarking or more complicated usage patterns.

Gradients

grad :: (Traversable f, Num a) => Direction -> (forall s. Mode s => f (AD s a) -> AD s a) -> f a -> f a

grad' :: (Traversable f, Num a) => Direction -> (forall s. Mode s => f (AD s a) -> AD s a) -> f a -> (a, f a)

Jacobians

jacobian :: (Traversable f, Traversable g, Num a) => Direction -> (forall s. Mode s => f (AD s a) -> g (AD s a)) -> f a -> g (f a)

jacobian' :: (Traversable f, Traversable g, Num a) => Direction -> (forall s. Mode s => f (AD s a) -> g (AD s a)) -> f a -> g (a, f a)

Derivatives

diff :: Num a => Direction -> (forall s. Mode s => AD s a -> AD s a) -> a -> a

diff' :: Num a => Direction -> (forall s. Mode s => AD s a -> AD s a) -> a -> (a, a)

Exposed Types

data Direction

Constructors

Forward

Reverse

Chain

Tower

Mixed

Instances

Bounded Direction

Enum Direction

Eq Direction

Ord Direction

Read Direction

Show Direction

Ix Direction