mlx/python
Awni Hannun 50f3535693
Use expand_dims / unflatten / etc in more places (#1696)
* use expand_dims / unflatten in a couple more places

* few more

* few more

* fix
2024-12-12 17:00:44 -08:00
..
mlx No reshapes in quantized embedding (#1682) 2024-12-09 18:57:38 -08:00
src Use expand_dims / unflatten / etc in more places (#1696) 2024-12-12 17:00:44 -08:00
tests Fix small sort with metal validation (#1695) 2024-12-12 09:21:45 -08:00