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 |
|
Jagrit Digani
|
776c3d226d
|
Convolution update (#651)
* Init steel conv and update Conv primitive
* Update slow CPU implementation to support flipping and input dilation winograd conv routing
Co-authored-by: Awni Hannun <awni@apple.com>
|
2024-02-28 20:11:16 -08:00 |
|
Awni Hannun
|
09b9275027
|
Make shape a tuple (#591)
* shape tuple
* also remove simplify from docs
* rebase
|
2024-01-30 13:11:01 -08:00 |
|
Zach Schillaci
|
5b9be57ac3
|
Add isort pre-commit and run (#68)
|
2023-12-08 11:31:47 -08:00 |
|
Awni Hannun
|
46a39e5b1f
|
copyright + ack
|
2023-11-30 11:12:53 -08:00 |
|
Jagrit Digani
|
e6306cfee9
|
jagrit's commit files
|
2023-11-29 10:52:08 -08:00 |
|