Module ad

tenferro_internal_ad_ops

Module ad

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.