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