zhangyiss/mlx
1
0
Fork 0
mirror of https://github.com/ml-explore/mlx.git synced 2025-06-24 09:21:16 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity
4f9b60dd53
mlx/python
History
Awni Hannun f76a49e555
ExpandDims primitive (#1687) 
* add squeeze primitive

* simplify squeeze, use in gather

* fix

* fix

* fix

* fix

* fix no cpu

* use squeeze in matmul and friends

* expand dims primitive

* comment
2024-12-10 16:39:07 -08:00
..
mlx No reshapes in quantized embedding (#1682) 2024-12-09 18:57:38 -08:00
src ExpandDims primitive (#1687) 2024-12-10 16:39:07 -08:00
tests ExpandDims primitive (#1687) 2024-12-10 16:39:07 -08:00