ConvTranspose2D implementation

[nickglazzard1959]

Hello all. I think I've seen some proposals for an implementation of this in MLX but I can't see it in the 0.10.0 release. It looks like the convolution implementation was overhauled recently, but this doesn't seem to have been added. Have I missed it somehow? If not, is there some way of using the existing MLX facilities to implement it (perhaps with some loss of efficiency, but to get the functionality)? Has anyone perhaps already done this? I don't think I have the skills needed to add it to MLX myself, so some higher level approach would be very helpful. I'd be grateful for any ideas.

[awni]

You can use mx.conv_general to implement the transposed convolution (which is the way we would do it if we added a ConvTranspose layer or op. I think there is still room to optimize the underlying implementation but it should not be horribly slow.

This is a useful page to see correspondence between conv and transposed conv.

[nickglazzard1959]

Thank you! That (mx.conv_general) had eluded me. That should do it though! Thanks for the very quick response.