t3toolbox.backend.common.numpy_scan

t3toolbox.backend.common.numpy_scan(f: Callable[[CarryType, Sequence[NDArray]], Tuple[CarryType, Sequence[NDArray]]], init: CarryType, xs: Sequence[Sequence[NDArray] | NDArray]) → Tuple[CarryType, Tuple[NDArray, Ellipsis]]

Similar to jax.lax.scan, except returns numpy arrays instead of jax arrays.