ConvMixer implementation.
See “Patches Are All You Need?”, submitted to ICLR’21 https://openreview.net/forum?id=TVHS5Y4dNvM