qutip_jax package

Diffrax ODE

class qutip_jax.DiffraxIntegrator(system, options)[source]

property options

Supported options by diffrax method:

dt0float, default=0.0001: Initial step size.
solver: AbstractSolver, default=Tsit5(),: ODE solver instance from diffrax.
stepsize_controller: AbstractStepSizeController, default=ConstantStepSize(): Step size controller from diffrax.
max_steps: int, default=100000: Maximum number of steps for the integration.