32668a7317
* add cholesky inv + tri inv * always run tri_inv on cpu * consistent naming
18 lines
200 B
ReStructuredText
18 lines
200 B
ReStructuredText
.. _linalg:
|
|
|
|
Linear Algebra
|
|
==============
|
|
|
|
.. currentmodule:: mlx.core.linalg
|
|
|
|
.. autosummary::
|
|
:toctree: _autosummary
|
|
|
|
inv
|
|
tri_inv
|
|
norm
|
|
cholesky
|
|
cholesky_inv
|
|
qr
|
|
svd
|