I've done some work in the past on NN representations and you actually can represent Conv and MatMul in more primitive ways. I ended up writing an IR called lo

I've done some work in the past on NN representations and you actually can represent Conv and MatMul in more primitive ways. I ended up writing an IR called loop_tool that exposes this stuff: https://github.com/facebookresearch/loop_tool/blob/main/python/nn.py#L76 https://github.com/facebookresearch/loop_tool/blob/main/pyth... The idea is basically this: https://news.ycombinator.com/item?id=28883086 https://news.ycombinator.com/item?id=28883086