Reorder top-level functions in lax.linalg, and add/expand docstrings.

PiperOrigin-RevId: 726535390
jax-ml · Feb 13, 2025 · 8d165ba · 8d165ba
1 parent 229aa65
commit 8d165ba
Show file tree

Hide file tree

Showing 3 changed files with 326 additions and 237 deletions.
diff --git a/docs/jax.lax.rst b/docs/jax.lax.rst
@@ -238,19 +238,24 @@ Linear algebra operators (jax.lax.linalg)
   :toctree: _autosummary
 
     cholesky
+    cholesky_update
     eig
     eigh
     hessenberg
-    lu
     householder_product
+    lu
+    lu_pivots_to_permutation
     qdwh
     qr
     schur
     svd
+    SvdAlgorithm
+    symmetric_product
     triangular_solve
     tridiagonal
     tridiagonal_solve
 
+
 Argument classes
 ----------------