zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-09-23 14:08:12 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
Files
4e1e9520e1fc5568feb536e2c692da81c7bef35b
mlx/python
History
Awni Hannun 4e1e9520e1 Flatten and unflatten (#1692) 
* flatten and unflatten

* fix grad

* fix shape infer

* use squeeze + unsqueeze in get_item
2024-12-11 21:51:37 -08:00
..
mlx
No reshapes in quantized embedding (#1682)
2024-12-09 18:57:38 -08:00
src
Flatten and unflatten (#1692)
2024-12-11 21:51:37 -08:00
tests
Flatten and unflatten (#1692)
2024-12-11 21:51:37 -08:00