mlx/python
Awni Hannun 195b429d99
Put along axis + fixe for partition grad (#1430)
* put along axis, fixes for partition grad

* zeros for arg reduce
2024-09-23 10:03:38 -07:00
..
mlx Faster RNN layers (#1419) 2024-09-17 06:04:19 -07:00
src Put along axis + fixe for partition grad (#1430) 2024-09-23 10:03:38 -07:00
tests Put along axis + fixe for partition grad (#1430) 2024-09-23 10:03:38 -07:00