13 Commits

Author	SHA1	Message	Date
Abe Leininger	3835a428c5	Adds nuclear norm support (#1894) ... * adjust norm unit test tolerance	2025-03-04 13:26:02 -08:00
Abe Leininger	a5ededf1c3	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-10 12:32:24 -08:00
Awni Hannun	516ded618b	Dynamic slicing (#1741) ... * dynamic slice and slice update * python bindings + tests + fix set item * fix compile issue * comment * fix jit	2025-01-07 14:02:16 -08:00
Awni Hannun	40c62c1321	Use int64 stride everywhere (#1671) ... * use int64 stride everywhere * fix ext * fix ext * more shape + cleanup * one more * few more	2024-12-09 11:09:02 -08:00
Kashif Rasul	3ddc07e936	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-22 12:18:48 -07:00
Nripesh Niketan	6af5ca35b2	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-17 13:12:43 -07:00
Aditya Dhulipala	e6b223df5f	Pinv (#875)	2024-08-27 23:06:12 -07:00
Luca Arnaboldi	b3ec792380	Implemented Cholesky on CPU (#1119)	2024-05-17 12:31:59 -07:00
nicolov	eaba55c9bf	Add matrix inversion primitive (#822)	2024-03-15 06:34:36 -07:00
nicolov	d0c544a868	Add SVD primitive (#809) ... Add SVD op using Accelerate's LAPACK following https://developer.apple.com/documentation/accelerate/ compressing_an_image_using_linear_algebra Co-authored-by: Nicolo Valigi <nvaligi@apple.com>	2024-03-12 12:30:11 -07:00
taher	077c1ee64a	QR factorization (#310) ... * add qr factorization --------- Co-authored-by: Awni Hannun <awni@apple.com>	2024-01-26 09:27:31 -08:00
Awni Hannun	f099ebe535	Multi output primitives (#330) ... * Multi-output primitives --------- Co-authored-by: Angelos Katharopoulos <a_katharopoulos@apple.com>	2024-01-08 16:39:08 -08:00
Gabrijel Boduljak	6b0d30bb85	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-26 19:42:04 -08:00