Module ad

Module ad 

Source

Functionsยง

einsum_frule
Stateless forward-mode rule (JVP) for einsum over dense primals.
einsum_rrule
Stateless reverse-mode rule (VJP) for einsum over dense primals.
solve_triangular_rrule
Stateless reverse-mode rule (VJP) for triangular solve.