L-electron-Rare/mlx
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
Files
195b429d9943da8db2e91a03786c7420d7b9bbed
mlx/python
T
History
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