mlx/python
Awni Hannun e4534dac17
Conv grad with groups + bugfix (#1449)
* fix bug in flipped conv with groups, start of grad for groups

* fix

* fix

* fix + test
2024-10-06 07:08:53 -07:00
..
mlx Add support for grouped 1D convolutions to the nn API (#1444) 2024-09-28 06:41:07 -07:00
src allow take to work with integer index (#1440) 2024-09-26 15:58:03 -07:00
tests Conv grad with groups + bugfix (#1449) 2024-10-06 07:08:53 -07:00