Edit docstring

lanl · Dec 1, 2023 · 5a45e13 · 5a45e13
1 parent eed9032
commit 5a45e13
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/scico/flax/inverse.py b/scico/flax/inverse.py
@@ -472,6 +472,8 @@ def power_iteration(A: LinearOperator, maxiter: int = 100):
 
     Compute largest eigenvalue of a diagonalizable
     :class:`LinearOperator` using power iteration.
+    The function makes use of lax functionality to
+    allow jitting and general jax function composition.
 
     Args:
         A: :class:`LinearOperator` used for computation. Must be