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.