Rifur13 c4a471c99d Add groups to Conv1d (#948) ... * Add conv1d grouped convs on CPU * Add GPU support * Parallelize inside metal kernel * clenaup * Update mlx/ops.cpp Co-authored-by: Awni Hannun <awni.hannun@gmail.com> * New unfold kernel + remove unused code * Remove copy and refactor * Update vjp and reuse steel gemm * Fixed groups on cpu * Fix metal validation --------- Co-authored-by: Awni Hannun <awni.hannun@gmail.com>		2024-04-27 06:24:57 -07:00
..
mlx	avgpool, not maxpool (#1002)	2024-04-17 08:26:22 -07:00
src	Add bitwise ops (#1037)	2024-04-26 22:03:42 -07:00
tests	Add groups to Conv1d (#948)	2024-04-27 06:24:57 -07:00