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