Integrators

Collection of symplectic (JAX composable) integrators

sympint.integrators.midpoint(H: Callable[[...], jax.Array], ns: int = 1) → Callable[[...], jax.Array]

Generate implicit midpoint integrator

Parameters:

• H (Callable[[Array, *Any], Array]) – Hamiltonian function H(q, p, dt, t, *args)

• ns (int, default=1) – number of Newton iteration steps

Returns:

integrator(qp, dt, t, *args)

Return type:

Callable[[Array, *Any], Array]

sympint.integrators.tao(H: Callable[[...], jax.Array], binding: float = 0.0) → Callable[[...], jax.Array]

Generate Tao integrator

Parameters:

• H (Callable[[Array, *Any], Array]) – Hamiltonian function H(q, p, dt, *args)

• binding (float, default=0.0) – binding factor

Returns:

integrator(qp, dt, *args)

Return type:

Callable[[Array, *Any], Array]