jax.lax.scan ============ .. currentmodule:: jax.lax .. autofunction:: scan