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