Jacket 3f7aba8498 Implement diagonal operator (#562) ...

* Implement diagonal operator

This implements mx.diagonal in operator level, inspired by
@ManishAradwad.

* added `mx.diag` with tests

* corrected few things

* nits in bindings

* updates to diag

---------

Co-authored-by: ManishAradwad <manisharadwad@gmail.com>
Co-authored-by: Awni Hannun <awni@apple.com>

2024-01-30 09:45:48 -08:00

..

mlx

Add more checks and clearer error messages to conv operations (#563)

2024-01-26 15:13:26 -08:00

src

Implement diagonal operator (#562)

2024-01-30 09:45:48 -08:00

tests

Implement diagonal operator (#562)

2024-01-30 09:45:48 -08:00