torch.matmul support

Thank you for your work. I would like to ask, in ViT, for layers like linear that have weights, it's easy to replace them. Are there examples of replacing operations like `torch.matmul`?
