Kashif Rasul
3ddc07e936
* 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>
21 lines
232 B
ReStructuredText
21 lines
232 B
ReStructuredText
.. _linalg:
|
|
|
|
Linear Algebra
|
|
==============
|
|
|
|
.. currentmodule:: mlx.core.linalg
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
inv
|
|
tri_inv
|
|
norm
|
|
cholesky
|
|
cholesky_inv
|
|
cross
|
|
qr
|
|
svd
|
|
eigvalsh
|
|
eigh
|