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