zoomy_core.fvm.jvp_numpy module

Module zoomy_core.fvm.jvp_numpy.

zoomy_core.fvm.jvp_numpy.fd_jvp(residual_fn, Q, V, eps=1e-07)

Fd jvp.

zoomy_core.fvm.jvp_numpy.analytic_source_jvp(runtime_model, symbolic_model, Q, Qaux, V, mesh, dt, include_chain_rule=True)

Analytic Jv for source residual S(Q, Qaux(Q)).

• Without chain rule: Jv = (dS/dQ) v

• With chain rule: Jv = (dS/dQ) v + (dS/dQaux) (dQaux/dQ v)