Angelos Katharopoulos d8fabaa12b Split multi output (#461) ...

* Multi-output split primitive
* Add the multi-output split to the ArrayIterator
* Add some grad tests for split

2024-01-16 13:33:55 -08:00

..

mlx

Add type hint for Module (#412)

2024-01-10 11:23:42 -08:00

src

implemented isposinf and isneginf in one PR (#470)

2024-01-16 06:48:07 -08:00

tests

Split multi output (#461)

2024-01-16 13:33:55 -08:00