make_fixed_point_layer

iskra.adjoint.make_fixed_point_layer(fn: Callable[[P], T], iterates: Sequence[tuple[int, int]] | tuple[int, int] = (0, 0), param_args: int | tuple[int, ...] = 1, fwd_method: Literal['fixed-point'] | Callable[[P], T] = 'fixed-point', fwd_max_iter: int | None = 100, fwd_error_metric: Literal['delta'] | int | None = 'delta', fwd_error_ord: int | float | Literal['fro', 'nuc'] = 2, fwd_abs_tol: float = 1e-06, fwd_rel_tol: float = 0.001, bwd_method: Literal['gmres'] = 'gmres', gmres_init: Tensor | None = None, callback_gmres_sol: Callable | None = None, bwd_max_iter: int = 500, bwd_abs_tol: float = 1e-06, bwd_rel_tol: float = 0.001, verbose: bool = False) → Callable[[P], T][SOURCE]