mlx/python
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