jaxprop.helpers_jax module

jaxprop.helpers_jax.format_list_80(items: list[str], width: int = 80) → str

jaxprop.helpers_jax.make_diffrax_adjoint(name: str, **kwargs)

jaxprop.helpers_jax.make_diffrax_solver(name: str, **kwargs)