Sobek: Streaming Equivariant Tensor Product Convolutions

Vladimir Chorošajev, Cédric Bény

Equivariant graph neural networks repeatedly apply edge-conditioned tensor-product convolutions over graph edges. Conventional implementations materialize edge-specific weights, me…