SEB in node classification

[gaarutyunov]

Hello,

First of all, great work!

I am implementing a node classification task using your model. It works perfectly with small tweaks and additions. However, I can't figure our how to implement SEB (sparse equivariant basis). Where do I need to add the torch.coalesce call?

Thanks in advance for your answer!