zoomy_jax.fvm.jvp_jax module

Module zoomy_jax.fvm.jvp_jax.

zoomy_jax.fvm.jvp_jax.analytic_source_jvp_jax(runtime_model, symbolic_model, Q, Qaux, V, mesh, dt, include_chain_rule=True)

Analytic source jvp jax.