mlx
mlx copied to clipboard
[Feature Request] Support convolution backward for groups > 1.
https://github.com/ml-explore/mlx/blob/03cf033f8299a5a694b31b650c6e03aa62c2a5b6/mlx/primitives.cpp#L903
Indeed, we should add this. Labeled as an enhancement.
This is now supported on main.