jax.lax.cumsum ============== .. currentmodule:: jax.lax .. autofunction:: cumsum