solvers

Solvers classes and functions.

Module Attributes

SPARSE_CYCLE_THRESHOLD

Evaluation-point count above which the single-sine-wave solver builds the dense power table.

Functions

compute_time_grid(t_span, t_eval, max_dt[, ...])	Construct integration steps aligned with the evaluation grid.
get_propagator(magnus_order)	Return a Magnus propagator closure of the requested order.
solve_time_segment_basic(solver, y0s, ...[, rng])	Solve a single time segment with uniform discretization.
solve_time_segment_sequence(solver, y0s, ...)	Solve a sequence of time segments and concatenate the results.
solve_time_segment_single_sine_wave(solver, ...)	Solve a constant-envelope single-sine segment.
step_solver_jax(propagator_fn, generator, ...)	Integrate the time evolution using JAX primitives and scans.
step_solver_numpy(propagator_fn, generator, ...)	Integrate the time evolution using NumPy-based matrix exponentials.
step_solver_parallel_jax(propagator_fn, ...)	Integrate the time evolution using an associative scan over JAX matrices.

Classes

Solver(static_hamiltonian[, ...])	Solver for the time-dependent Schrödinger equation.
SolverResult(ts, ys, *, time_axis[, shot_axis])	Container for solver outputs that preserves array backend compatibility.