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
..
cpp	Update pre-commit hooks (#984)	2024-04-11 07:27:53 -07:00
numpy	Add isort pre-commit and run (#68)	2023-12-08 11:31:47 -08:00
python	Add groups to Conv1d (#948)	2024-04-27 06:24:57 -07:00