zoomy_jax.fvm.ode module

Module zoomy_jax.fvm.ode.

zoomy_jax.fvm.ode.RK1(func, Q, Qaux, param, dt, func_jac=None, func_bc=None)

RK1.

zoomy_jax.fvm.ode.RK2(func, Q, Qaux, param, dt, func_jac=None, func_bc=None)

SSP-RK2 (Heun’s method). JIT-compatible.

zoomy_jax.fvm.ode.RK3(func, Q, Qaux, param, dt, func_jac=None, func_bc=None)

SSP-RK3. JIT-compatible.

zoomy_jax.fvm.ode.RKimplicit(func, Q, Qaux, param, dt, func_jac=None, func_bc=None)

implicit euler