Replacing torch.kron with fastkron
RaulPPelaez opened this issue · 1 comments
RaulPPelaez commented
Hi, thanks for your nice contribution!
Are there any plans to write a replacement for torch.kron
?
Either in torch itself or providing a custom CUDA/C++ extension through this repo.
That would entail writing a backwards pass for fastkron as well.
abhijangda commented
Hi Raul,
Yes, the plan is to provide a Python extension that does Kronecker Matrix-Matrix Multiplication. However, I think that requires a few months. I am not sure about the backward pass yet. I think that is more evolved.