Jagrit Digani 5ad133f8bb No copy gems (#801) ...

* Enable collapsing batch dims in gemm
* Update gemm to only make copies when neither of the last 2 axes are contiguous
* Update addmm to support gemv shapes
* Update addmm to support irregular batch strides
* Update tests

2024-03-12 13:13:41 -07:00

..

mlx

fix modules with dict (#819)

2024-03-12 08:54:06 -07:00

src

Add SVD primitive (#809)

2024-03-12 12:30:11 -07:00

tests

No copy gems (#801)

2024-03-12 13:13:41 -07:00