zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-27 03:11:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
6871e2eeb7
mlx/docs/src/python/linalg.rst

28 lines
311 B
ReStructuredText
Raw Normal View History

linalg.norm (#187) * implemented vector_norm in cpp added linalg to mlx * implemented vector_norm python binding * renamed vector_norm to norm, implemented norm without provided ord * completed the implementation of the norm * added tests * removed unused import in linalg.cpp * updated python bindings * added some tests for python bindings * handling inf, -inf as numpy does, more extensive tests of compatibility with numpy * added better docs and examples * refactored mlx.linalg.norm bindings * reused existing util for implementation of linalg.norm * more tests * fixed a bug with no ord and axis provided * removed unused imports * some style and API consistency updates to linalg norm * remove unused includes * fix python tests * fixed a bug with frobenius norm of a complex-valued matrix * complex for vector too --------- Co-authored-by: Awni Hannun <awni@apple.com>
2023-12-27 11:42:04 +08:00
.. _linalg:
Linear Algebra
==============
.. currentmodule:: mlx.core.linalg
CPU LU factorization and linear solvers (#1451) * linalg solve backend * nits * more nits + fix * luf primitive and lu, solve, and solve_triangular backends * changes / nits --------- Co-authored-by: Awni Hannun <awni@apple.com>
2025-02-11 04:32:24 +08:00
.. autosummary::
:toctree: _autosummary
linalg.norm (#187) * implemented vector_norm in cpp added linalg to mlx * implemented vector_norm python binding * renamed vector_norm to norm, implemented norm without provided ord * completed the implementation of the norm * added tests * removed unused import in linalg.cpp * updated python bindings * added some tests for python bindings * handling inf, -inf as numpy does, more extensive tests of compatibility with numpy * added better docs and examples * refactored mlx.linalg.norm bindings * reused existing util for implementation of linalg.norm * more tests * fixed a bug with no ord and axis provided * removed unused imports * some style and API consistency updates to linalg norm * remove unused includes * fix python tests * fixed a bug with frobenius norm of a complex-valued matrix * complex for vector too --------- Co-authored-by: Awni Hannun <awni@apple.com>
2023-12-27 11:42:04 +08:00
Add missing docs for some ops (#1110)
2024-05-14 21:09:05 +08:00
inv
CPU mx.linalg.cholesky_inverse and mx.linalg.tri_inv (#1307) * add cholesky inv + tri inv * always run tri_inv on cpu * consistent naming
2024-08-09 06:18:02 +08:00
tri_inv
linalg.norm (#187) * implemented vector_norm in cpp added linalg to mlx * implemented vector_norm python binding * renamed vector_norm to norm, implemented norm without provided ord * completed the implementation of the norm * added tests * removed unused import in linalg.cpp * updated python bindings * added some tests for python bindings * handling inf, -inf as numpy does, more extensive tests of compatibility with numpy * added better docs and examples * refactored mlx.linalg.norm bindings * reused existing util for implementation of linalg.norm * more tests * fixed a bug with no ord and axis provided * removed unused imports * some style and API consistency updates to linalg norm * remove unused includes * fix python tests * fixed a bug with frobenius norm of a complex-valued matrix * complex for vector too --------- Co-authored-by: Awni Hannun <awni@apple.com>
2023-12-27 11:42:04 +08:00
norm
Some fixes in docs (#1141) * fixes in docs * nit
2024-05-21 02:51:47 +08:00
cholesky
CPU mx.linalg.cholesky_inverse and mx.linalg.tri_inv (#1307) * add cholesky inv + tri inv * always run tri_inv on cpu * consistent naming
2024-08-09 06:18:02 +08:00
cholesky_inv
feat: add cross_product (#1252) * feat: add cross_product * lint * python binding * refactor: Improve error message for cross_product function * refactor: more close to numpy cross product * refactor: improve error message for cross_product function * finish * fix acks * allow old numpy * doc --------- Co-authored-by: Awni Hannun <awni@apple.com>
2024-09-18 04:12:43 +08:00
cross
QR factorization (#310) * add qr factorization --------- Co-authored-by: Awni Hannun <awni@apple.com>
2024-01-27 01:27:31 +08:00
qr
Add missing docs for some ops (#1110)
2024-05-14 21:09:05 +08:00
svd
non-symmetric eig and eigh (#2188)
2025-05-16 04:01:44 +08:00
eigvals
eig
Eigenvalues and eigenvectors (#1334) * initial eigvalsh * add compute_vectors * add compute_vectors_ * return a pair * add eigh to return only eigenvectors * fixed typo * merge merge Eighvalsh and Eigh into a single primitive * use the same primate with the flag * fix primatives * use MULTI * fix eval_gpu * fix decleration * rename EighPrimitive to Eigh * tests * tests * fix rebase and format * cleanup lapack * format * add cblas.h --------- Co-authored-by: Awni Hannun <awni@apple.com>
2024-10-23 03:18:48 +08:00
eigvalsh
eigh
CPU LU factorization and linear solvers (#1451) * linalg solve backend * nits * more nits + fix * luf primitive and lu, solve, and solve_triangular backends * changes / nits --------- Co-authored-by: Awni Hannun <awni@apple.com>
2025-02-11 04:32:24 +08:00
lu
lu_factor
add pinv to doc (#2020)
2025-03-31 06:54:18 +08:00
pinv
CPU LU factorization and linear solvers (#1451) * linalg solve backend * nits * more nits + fix * luf primitive and lu, solve, and solve_triangular backends * changes / nits --------- Co-authored-by: Awni Hannun <awni@apple.com>
2025-02-11 04:32:24 +08:00
solve
solve_triangular
Reference in New Issue Copy Permalink