mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

DOC: add missing linalg functionality to docs

Browse Source
This commit is contained in:
Jake VanderPlas 2022-03-15 09:55:59 -07:00
parent 98ad016794
commit c66f5dda60
3 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
docs/jax.lax.rst
View File

@ -207,8 +207,10 @@ Linear algebra operators (jax.lax.linalg)
lu
qdwh
qr
schur
svd
triangular_solve
tridiagonal_solve
Argument classes
----------------

4
docs/jax.scipy.rst
View File

@ -26,17 +26,21 @@ jax.scipy.linalg
cholesky
det
eigh
eigh_tridiagonal
expm
expm_frechet
inv
lu
lu_factor
lu_solve
polar
polar_unitary
qr
schur
sqrtm
solve
solve_triangular
sqrtm
svd
tril
triu

2
jax/_src/lax/linalg.py
View File

@ -1461,7 +1461,7 @@ xla.register_translation(tridiagonal_solve_p, xla.lower_fun(
def tridiagonal_solve(dl, d, du, b):
r"""Computes the solution of a tridiagonal linear system.
This function computes the solution of a tridiagonal linear system::
This function computes the solution of a tridiagonal linear system:
.. math::
A . X = B