flow.iterate_to_steady
flow.iterate_to_steady(step_fn, theta, f0, options)Pseudo-time march to the fixed point, without building a tape.
Parameters
| Name | Type | Description | Default |
|---|---|---|---|
| step_fn | Callable | (f, theta) -> f. |
required |
| theta | Any | The differentiable parameters. | required |
| f0 | jax.Array | Initial populations. | required |
| options | SteadyOptions | Convergence settings. | required |
Returns
| Name | Type | Description |
|---|---|---|
| jax.Array | The converged populations. |