Portability | GHC only |
---|---|
Stability | experimental |
Maintainer | ekmett@gmail.com |
Safe Haskell | None |
Allows the choice of AD Mode
to be specified at the term level for
benchmarking or more complicated usage patterns.
- grad :: (Traversable f, Num a) => Direction -> (forall t. Mode t => f t -> t) -> f a -> f a
- grad' :: (Traversable f, Num a) => Direction -> (forall t. Mode t => f t -> t) -> f a -> (a, f a)
- jacobian :: (Traversable f, Traversable g, Num a) => Direction -> (forall t. Mode t => f t -> g t) -> f a -> g (f a)
- jacobian' :: (Traversable f, Traversable g, Num a) => Direction -> (forall t. Mode t => f t -> g t) -> f a -> g (a, f a)
- diff :: Num a => Direction -> (forall t. Mode t => t -> t) -> a -> a
- diff' :: Num a => Direction -> (forall t. Mode t => t -> t) -> a -> (a, a)
- data Direction
Gradients
grad' :: (Traversable f, Num a) => Direction -> (forall t. Mode t => f t -> t) -> f a -> (a, f a)Source
Jacobians
jacobian :: (Traversable f, Traversable g, Num a) => Direction -> (forall t. Mode t => f t -> g t) -> f a -> g (f a)Source
jacobian' :: (Traversable f, Traversable g, Num a) => Direction -> (forall t. Mode t => f t -> g t) -> f a -> g (a, f a)Source