Incompatible with einops

[CreamNuts]

Thanks for your nice project. But it is not compatible with einops.
Einops is convenient for pytorch user, and I think many people use it.
I hope it works on einops, too. Thank you.

[rentruewang]

Good point. I'll be working on it. Feel free to add more feature requests!

[rentruewang]

Ok. After quickly skimming over einops's source code, I noticed that they call torch.* functions behind the scenes. That means their functions will directly work with Koila, because Koila uses PyTorch's extension API.

With that being said, not all torch.* functions are implemented yet, and it probably requires some testing. So I'll leave it open for now, and see what I can do when I have some time.

[CreamNuts]

I'm looking forward to implementing it as soon as possible.