mlx/python
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
..
mlx fix submodule stubs (#1492) 2024-10-15 16:23:37 -07:00
src Eigenvalues and eigenvectors (#1334) 2024-10-22 12:18:48 -07:00
tests Eigenvalues and eigenvectors (#1334) 2024-10-22 12:18:48 -07:00