A JIT kernel generator for the Clebsch-Gordan tensor product reports up to 1.3x speedup over NVIDIA cuEquivariance, over 10x over e3nn, and a 6.2x FP64 inference speedup for the MACE model over its unoptimized baseline.
Brown- ing, and Michele Ceriotti
1 Pith paper cite this work. Polarity classification is still indexing.
1
Pith paper citing it
citation-role summary
background 1
citation-polarity summary
fields
cs.LG 1years
2025 1verdicts
CONDITIONAL 1roles
background 1polarities
unclear 1representative citing papers
citing papers explorer
-
An Efficient Sparse Kernel Generator for O(3)-Equivariant Deep Networks
A JIT kernel generator for the Clebsch-Gordan tensor product reports up to 1.3x speedup over NVIDIA cuEquivariance, over 10x over e3nn, and a 6.2x FP64 inference speedup for the MACE model over its unoptimized baseline.