bellini.api.functional.functional_for

bellini.api.functional.functional_for(f, init, xs, length=None)[source]

A wrapper for numpyro’s scan primative that also contains a non-numpyro counterpart implementation with similar functionality. See jax.lax.scan for detailed code signature.