Fast contracted Clebsch--Gordan tensor products for equivariant graph neural networks
Abstract
We present an algorithm for evaluating contracted Clebsch--Gordan tensor products in -equivariant machine learning potentials at fixed Canonical Polyadic (CP) rank. Mapping the angular integral to a structured Gauss--Legendre and Fourier tensor-product grid decouples the radial channel contractions from the angular transforms. The antisymmetric parity-odd Clebsch--Gordan channels, unreachable by the symmetric pointwise product on a scalar grid, are recovered through the surface-curl pairing , the spherical Poisson bracket, which supplies the angular momentum on the grid while preserving rotational equivariance. The construction extends to parity-aware equivariant message passing in atomic-cluster-expansion-style architectures and is verified by direct numerical quadrature. The full uncontracted Clebsch--Gordan tensor product remains subject to the output-size lower bound. A benchmark shows wall-clock scaling empirically as across the practical range. For the on-site contraction this is pre-asymptotic, giving way to at large . For message passing it is structural and the runtime is memory-bandwidth bound on -sized grid tensors.
Keywords
Cite
@article{arxiv.2605.15073,
title = {Fast contracted Clebsch--Gordan tensor products for equivariant graph neural networks},
author = {Anton Bochkarev and Yury Lysogorskiy and Ralf Drautz},
journal= {arXiv preprint arXiv:2605.15073},
year = {2026}
}