MathiesW

Hi @patrick-kidger, thanks for the quick response. You may be right here, I have no idea what happens under the hood. I just noticed that the convolution operation in [Haiku](https://dm-haiku.readthedocs.io/en/latest/api.html#haiku.ConvND)...